Safe Haskell | None |
---|---|
Language | Haskell2010 |
- data TensorFlowException = TensorFlowException Code Text
- data Session
- withSession :: (SessionOptions -> IO ()) -> ((IO () -> IO ()) -> Session -> IO a) -> IO a
- extendGraph :: Session -> GraphDef -> IO ()
- run :: Session -> [(ByteString, TensorData)] -> [ByteString] -> [ByteString] -> IO [TensorData]
- data TensorData = TensorData {
- tensorDataDimensions :: [Int64]
- tensorDataType :: !DataType
- tensorDataBytes :: !(Vector Word8)
- setSessionConfig :: ConfigProto -> SessionOptions -> IO ()
- setSessionTarget :: ByteString -> SessionOptions -> IO ()
- getAllOpList :: IO ByteString
- useProtoAsVoidPtrLen :: (Message msg, Integral c, Show c, Bits c) => msg -> (Ptr b -> c -> IO a) -> IO a
Documentation
data TensorFlowException
TensorFlowException Code Text |
data Session
:: (SessionOptions -> IO ()) | |
-> ((IO () -> IO ()) -> Session -> IO a) | The action can spawn concurrent tasks which will be canceled before withSession returns. |
-> IO a |
Runs the given action after creating a session with options populated by the given optionSetter.
extendGraph :: Session -> GraphDef -> IO ()
:: Session | |
-> [(ByteString, TensorData)] | Feeds. |
-> [ByteString] | Fetches. |
-> [ByteString] | Targets. |
-> IO [TensorData] |
data TensorData
All of the data needed to represent a tensor.
TensorData | |
|
setSessionConfig :: ConfigProto -> SessionOptions -> IO ()
setSessionTarget :: ByteString -> SessionOptions -> IO ()
Returns the serialized OpList of all OpDefs defined in this address space.