| Safe Haskell | None |
|---|---|
| Language | Haskell2010 |
TensorFlow.Internal.FFI
Contents
- 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, Num c) => msg -> (Ptr b -> c -> IO a) -> IO a
Documentation
data TensorFlowException Source
Constructors
| TensorFlowException Code Text |
Arguments
| :: (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 () Source
Arguments
| :: Session | |
| -> [(ByteString, TensorData)] | Feeds. |
| -> [ByteString] | Fetches. |
| -> [ByteString] | Targets. |
| -> IO [TensorData] |
data TensorData Source
All of the data needed to represent a tensor.
Constructors
| TensorData | |
Fields
| |
Instances
setSessionConfig :: ConfigProto -> SessionOptions -> IO () Source
setSessionTarget :: ByteString -> SessionOptions -> IO () Source
getAllOpList :: IO ByteString Source
Returns the serialized OpList of all OpDefs defined in this address space.