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 0 |
major version More... | |
#define | MXNET_MINOR 11 |
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 0 |
major version
#define MXNET_MAKE_VERSION | ( | major, | |
minor, | |||
patch | |||
) | ((major)*10000 + (minor)*100 + patch) |
helper for making version number
#define MXNET_MINOR 11 |
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