mxnet
Classes | Namespaces | Macros | Typedefs
base.h File Reference
#include <dmlc/base.h>
#include <dmlc/io.h>
#include <dmlc/type_traits.h>
#include <dmlc/parameter.h>
#include <mshadow/tensor.h>
#include <nnvm/op.h>
#include <nnvm/tuple.h>
#include <nnvm/symbolic.h>
#include <string>
Include dependency graph for base.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  mxnet::Context
 Context information about the execution environment. More...
 
struct  mxnet::RunContext
 execution time context. The information needed in runtime for actual execution. More...
 

Namespaces

 mxnet
 namespace of mxnet
 

Macros

#define MXNET_USE_OPENCV   1
 whether to use opencv support More...
 
#define MXNET_USE_CUDNN   MSHADOW_USE_CUDNN
 whether to use cuda support More...
 
#define MXNET_USE_CUSOLVER   MSHADOW_USE_CUSOLVER
 whether to use cusolver library More...
 
#define MXNET_GPU_NOT_ENABLED_ERROR   "GPU is not enabled"
 Error message for using gpu when MXNET_USE_CUDA==0. More...
 
#define MXNET_API
 define compatible keywords in g++ Used to support g++-4.6 and g++4.7 More...
 
#define MXNET_PREDICT_ONLY   0
 define prediction only More...
 
#define PROFILER_MESSAGE(msg)   nullptr
 define operator message for profiler More...
 
#define MXNET_MAJOR   1
 major version More...
 
#define MXNET_MINOR   1
 minor version More...
 
#define MXNET_PATCH   0
 patch version More...
 
#define MXNET_VERSION   (MXNET_MAJOR*10000 + MXNET_MINOR*100 + MXNET_PATCH)
 mxnet version More...
 
#define MXNET_MAKE_VERSION(major, minor, patch)   ((major)*10000 + (minor)*100 + patch)
 helper for making version number More...
 
#define PROFILER_MESSAGE_FUNCNAME   PROFILER_MESSAGE(__FUNCTION__)
 define function name as profiler message More...
 

Typedefs

typedef mshadow::cpu mxnet::cpu
 mxnet cpu More...
 
typedef mshadow::gpu mxnet::gpu
 mxnet gpu More...
 
typedef mshadow::index_t mxnet::index_t
 index type usually use unsigned More...
 
typedef mshadow::default_real_t mxnet::real_t
 data type that will be used to store ndarray More...
 
using mxnet::TShape = nnvm::TShape
 Shape data structure used to record shape information. More...
 
using mxnet::Op = nnvm::Op
 operator structure from NNVM More...
 

Macro Definition Documentation

#define MXNET_API

define compatible keywords in g++ Used to support g++-4.6 and g++4.7

define dllexport for Visual Studio

#define MXNET_GPU_NOT_ENABLED_ERROR   "GPU is not enabled"

Error message for using gpu when MXNET_USE_CUDA==0.

#define MXNET_MAJOR   1

major version

#define MXNET_MAKE_VERSION (   major,
  minor,
  patch 
)    ((major)*10000 + (minor)*100 + patch)

helper for making version number

#define MXNET_MINOR   1

minor version

#define MXNET_PATCH   0

patch version

#define MXNET_PREDICT_ONLY   0

define prediction only

#define MXNET_USE_CUDNN   MSHADOW_USE_CUDNN

whether to use cuda support

whether to use cudnn library for convolution

#define MXNET_USE_CUSOLVER   MSHADOW_USE_CUSOLVER

whether to use cusolver library

#define MXNET_USE_OPENCV   1

whether to use opencv support

#define MXNET_VERSION   (MXNET_MAJOR*10000 + MXNET_MINOR*100 + MXNET_PATCH)

mxnet version

#define PROFILER_MESSAGE (   msg)    nullptr

define operator message for profiler

#define PROFILER_MESSAGE_FUNCNAME   PROFILER_MESSAGE(__FUNCTION__)

define function name as profiler message