mxnet
|
configuation of mxnet as well as basic data structure. More...
#include <dmlc/base.h>
#include <dmlc/io.h>
#include <dmlc/type_traits.h>
#include <dmlc/parameter.h>
#include <mshadow/tensor.h>
#include <string>
Go to the source code of this file.
Classes | |
struct | mxnet::Context |
Context information about the execution enviroment. 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_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... | |
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... | |
typedef mshadow::TShape | mxnet::TShape |
dynamic shape type More... | |
typedef mshadow::TBlob | mxnet::TBlob |
storage container type More... | |
configuation of mxnet as well as basic data structure.
Copyright (c) 2015 by Contributors
#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_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_OPENCV 1 |
whether to use opencv support