Copyright | (c) 2016-2017 Tao He |
---|---|
License | MIT |
Maintainer | sighingnow@gmail.com |
Safe Haskell | Safe |
Language | Haskell2010 |
Types in MXNet and NNVM.
- type NNUInt = CUInt
- type MXUInt = CUInt
- type MXFloat = CFloat
- data NDArrayHandle
- data FunctionHandle
- type AtomicSymbolCreator = OpHandle
- data SymbolHandle
- data AtomicSymbolHandle
- data ExecutorHandle
- data DataIterCreator
- data DataIterHandle
- data KVStoreHandle
- data RecordIOHandle
- data RtcHandle
- data PredictorHandle
- data NDListHandle
- type OpHandle = Ptr ()
- data GraphHandle
- data ExecutorMonitorCallback
- data CustomOpPropCreator
- type MXKVStoreUpdater = Int -> NDArrayHandle -> NDArrayHandle -> Ptr () -> IO Int
- type MXKVStoreServerController = Int -> Ptr CChar -> Ptr () -> IO Int
Type alias
Handlers and Creators
data NDArrayHandle Source #
Handle to NDArray.
data FunctionHandle Source #
Handle to a mxnet narray function that changes NDArray.
type AtomicSymbolCreator = OpHandle Source #
Handle to a function that takes param and creates symbol.
data SymbolHandle Source #
Handle to a symbol that can be bind as operator.
data ExecutorHandle Source #
Storable ExecutorHandle Source # | Handle to an Executor. |
Handlers in predict API
Handlers in nnvm
Callback types
data ExecutorMonitorCallback Source #
Callback: ExecutorMonitorCallback.
data CustomOpPropCreator Source #
Callback: CustomOpPropCreator.
type MXKVStoreUpdater Source #
= Int | The key. |
-> NDArrayHandle | The pushed value on the key. |
-> NDArrayHandle | The value stored on local on the key. |
-> Ptr () | The additional handle to the updater. |
-> IO Int |
Callback: MXKVStoreUpdater, user-defined updater for the kvstore.