mxnet-nnvm-0.1.0.0: NNVM interface in Haskell.

Copyright(c) 2016 Tao He
LicenseMIT
Maintainersighingnow@gmail.com
Safe HaskellSafe
LanguageHaskell2010

MXNet.NNVM.Base

Contents

Description

Interfaces in core module of NNVM.

Synopsis

Re-export data type definitions

type NNUInt = CUInt Source #

MXUint type alias.

data OpHandle Source #

Handle to a function that takes param and creates symbol.

Re-export functions.

nnAddControlDeps Source #

Arguments

:: SymbolHandle

The symbol to add dependency edges on.

-> SymbolHandle

The source handles.

-> IO Int 

nnSymbolGetAttr Source #

Arguments

:: SymbolHandle

symbol handle

-> String

key

-> IO (Int, String, Int) 

nnSymbolSetAttrs Source #

Arguments

:: SymbolHandle 
-> NNUInt 
-> [String]

attribute keys

-> [String]

attribute values

-> IO Int 

nnGraphGetSymbol Source #

Arguments

:: GraphHandle

the graph handle.

-> IO (Int, SymbolHandle)