MXNet.Core.NNVM.Internal

Interfaces in core module of NNVM.

Re-export data type definitions

type NNUInt = CUInt

Handle size_t type.

NNUint type alias.

type OpHandle = Ptr ()

Handle to a function that takes param and creates symbol.

data SymbolHandle

Handle to a symbol that can be bind as operator.

Instances

Storable SymbolHandle

Methods

sizeOf :: SymbolHandle -> Int

alignment :: SymbolHandle -> Int

peekElemOff :: Ptr SymbolHandle -> Int -> IO SymbolHandle

pokeElemOff :: Ptr SymbolHandle -> Int -> SymbolHandle -> IO ()

peekByteOff :: Ptr b -> Int -> IO SymbolHandle

pokeByteOff :: Ptr b -> Int -> SymbolHandle -> IO ()

peek :: Ptr SymbolHandle -> IO SymbolHandle

poke :: Ptr SymbolHandle -> SymbolHandle -> IO ()

data GraphHandle

Handle to Graph.

Instances

Storable GraphHandle

Methods

sizeOf :: GraphHandle -> Int

alignment :: GraphHandle -> Int

peekElemOff :: Ptr GraphHandle -> Int -> IO GraphHandle

pokeElemOff :: Ptr GraphHandle -> Int -> GraphHandle -> IO ()

peekByteOff :: Ptr b -> Int -> IO GraphHandle

pokeByteOff :: Ptr b -> Int -> GraphHandle -> IO ()

peek :: Ptr GraphHandle -> IO GraphHandle

poke :: Ptr GraphHandle -> GraphHandle -> IO ()

Re-export functions.

nnAPISetLastError :: String -> IO ()

Set the last error message needed by C API.

nnGetLastError :: IO String

Return str message of the last error.

nnListAllOpNames :: IO (Int, [String])

List all the available operator names, include entries.

nnGetOpHandle

Arguments

:: String

The name of the operator.

-> IO (Int, OpHandle)

Get operator handle given name.

nnListUniqueOps :: IO (Int, [OpHandle])

List all the available operators.

nnGetOpInfo :: OpHandle -> IO (Int, String, String, NNUInt, [String], [String], [String], String)

Get the detailed information about atomic symbol.

nnSymbolCreateAtomicSymbol

Arguments

:: OpHandle

The operator handle.

-> NNUInt

The number of parameters.

-> [String]

The keys to the params.

-> [String]

The values to the params.

-> IO (Int, SymbolHandle)

Create an AtomicSymbol functor.

nnSymbolCreateVariable

Arguments

:: String

The name of the variable.

-> IO (Int, SymbolHandle)

Create a Variable Symbol.

nnSymbolCreateGroup

Arguments

:: NNUInt

Number of symbols to be grouped.

-> [SymbolHandle]

Array of symbol handles.

-> IO (Int, SymbolHandle)

Create a Symbol by grouping list of symbols together.

nnAddControlDeps

Arguments

:: SymbolHandle

The symbol to add dependency edges on.

-> SymbolHandle

The source handles.

-> IO Int

Add src_dep to the handle as control dep.

nnSymbolFree :: SymbolHandle -> IO Int

Free the symbol handle.

nnSymbolCopy :: SymbolHandle -> IO (Int, SymbolHandle)

Copy the symbol to another handle.

nnSymbolPrint :: SymbolHandle -> IO (Int, String)

Print the content of symbol, used for debug.

nnSymbolGetAttr

Arguments

:: SymbolHandle

symbol handle

-> String

key

-> IO (Int, String, Int)

Get string attribute from symbol.

nnSymbolSetAttrs

Arguments

:: SymbolHandle

-> NNUInt

-> [String]

attribute keys

-> [String]

attribute values

-> IO Int

Set string attribute from symbol.

nnSymbolListAttrs :: SymbolHandle -> Int -> IO (Int, [String])

Get all attributes from symbol, including all descendents.

nnSymbolListInputVariables :: SymbolHandle -> Int -> IO (Int, [SymbolHandle])

List inputs variables in the symbol.

nnSymbolListInputNames :: SymbolHandle -> Int -> IO (Int, [String])

List input names in the symbol.

nnSymbolListOutputNames :: SymbolHandle -> IO (Int, [String])

List returns names in the symbol.

nnSymbolGetInternals :: SymbolHandle -> IO (Int, SymbolHandle)

Get a symbol that contains all the internals.

nnSymbolGetOutput :: SymbolHandle -> NNUInt -> IO (Int, SymbolHandle)

Get index-th outputs of the symbol.

nnSymbolCompose

Arguments

:: SymbolHandle

Creator/Handler of the OP.

-> String

-> [String]

-> [SymbolHandle]

-> IO Int

Invoke a nnvm op and imperative function.

nnGraphCreate :: SymbolHandle -> IO (Int, GraphHandle)

Create a graph handle from symbol.

nnGraphFree :: GraphHandle -> IO Int

Free the graph handle.

nnGraphGetSymbol

Arguments

:: GraphHandle

the graph handle.

-> IO (Int, SymbolHandle)

Get a new symbol from the graph.

nnGraphSetJSONAttr :: GraphHandle -> String -> String -> IO Int

Get Set a attribute in json format.

nnGraphGetJSONAttr :: SymbolHandle -> String -> IO (Int, String, Int)

Get a serialized attrirbute from graph.

nnGraphSetNodeEntryListAttr_ :: GraphHandle -> String -> SymbolHandle -> IO Int

Set a attribute whose type is std::vectorNodeEntry in c++.

nnGraphApplyPasses :: GraphHandle -> NNUInt -> [String] -> IO (Int, GraphHandle)

Apply passes on the src graph.