MXNet.NNVM.Base

Interfaces in core module of NNVM.

Re-export data type definitions

type NNUInt = CUInt

MXUint type alias.

data OpHandle

Handle to a function that takes param and creates symbol.

Instances

Storable OpHandle

Methods

sizeOf :: OpHandle -> Int

alignment :: OpHandle -> Int

peekElemOff :: Ptr OpHandle -> Int -> IO OpHandle

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

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

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

peek :: Ptr OpHandle -> IO OpHandle

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

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 ()

nnGetLastError :: IO String

nnListAllOpNames :: IO (Int, NNUInt, [String])

nnGetOpHandle :: String -> IO (Int, OpHandle)

nnListUniqueOps :: IO (Int, NNUInt, [OpHandle])

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

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

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

nnSymbolCreateGroup :: NNUInt -> [SymbolHandle] -> IO (Int, SymbolHandle)

nnAddControlDeps

Arguments

:: SymbolHandle

The symbol to add dependency edges on.

-> SymbolHandle

The source handles.

-> IO Int

nnSymbolFree :: SymbolHandle -> IO Int

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

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

nnSymbolGetAttr

Arguments

:: SymbolHandle

symbol handle

-> String

key

-> IO (Int, String, Int)

nnSymbolSetAttrs

Arguments

:: SymbolHandle

-> NNUInt

-> [String]

attribute keys

-> [String]

attribute values

-> IO Int

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

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

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

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

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

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

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

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

nnGraphFree :: GraphHandle -> IO Int

nnGraphGetSymbol

Arguments

:: GraphHandle

the graph handle.

-> IO (Int, SymbolHandle)

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

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

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

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