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	MXNET_MAJOR   1
	major version More...
#define	MXNET_MINOR   4
	minor version More...
#define	MXNET_PATCH   1
	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::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   4

minor version

#define MXNET_PATCH   1

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

define function name as profiler message