mxnet
|
configuration of MXNet as well as basic data structure. More...
#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"
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 dllexport for Visual Studio More... | |
#define | MXNET_PREDICT_ONLY 0 |
define prediction only More... | |
#define | MXNET_MAJOR 2 |
major version More... | |
#define | MXNET_MINOR 0 |
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::lapack_index_t | mxnet::lapack_index_t |
index type for blas library. 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... | |
configuration of MXNet as well as basic data structure.
#define MXNET_API |
define dllexport for Visual Studio
#define MXNET_MAJOR 2 |
major version
#define MXNET_MAKE_VERSION | ( | major, | |
minor, | |||
patch | |||
) | ((major)*10000 + (minor)*100 + patch) |
helper for making version number
#define MXNET_MINOR 0 |
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