|
mxnet
|
#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>

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... | |
| #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
1.8.11