mxnet
Classes | Namespaces | Macros | Typedefs
base.h File Reference
#include "dmlc/base.h"
#include <string>
#include "dmlc/io.h"
#include "dmlc/type_traits.h"
#include "dmlc/parameter.h"
#include "mshadow/tensor.h"
#include "nnvm/op.h"
#include "nnvm/symbolic.h"
#include "libinfo.h"
#include "tuple.h"
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...
 
class  mxnet::GPUAuxStream
 Holds an auxiliary mshadow gpu stream that can be synced with a primary stream. More...
 
class  mxnet::SyncedGPUAuxStream
 Provides automatic coordination of an auxilary stream with a primary one. This object, upon construction, prepares an aux stream for use by syncing it with enqueued primary-stream work. Object destruction will sync again so future primary-stream work will wait on enqueued aux-stream work. If MXNET_GPU_WORKER_NSTREAMS == 1, then this defaults simply: the primary stream will equal the aux stream and the syncs will be executed as nops. See ./src/operator/cudnn/cudnn_convolution-inl.h for a usage example. More...
 
struct  mxnet::RunContext
 execution time context. The information needed in runtime for actual execution. More...
 

Namespaces

 mxnet
 namespace of mxnet
 

Macros

#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 MXNET_MAJOR   1
 major version More...
 
#define MXNET_MINOR   7
 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   (__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::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_MAJOR   1

major version

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

helper for making version number

#define MXNET_MINOR   7

minor version

#define MXNET_PATCH   0

patch version

#define MXNET_PREDICT_ONLY   0

define prediction only

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

mxnet version

#define PROFILER_MESSAGE_FUNCNAME   (__FUNCTION__)

define function name as profiler message