#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.
|
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...
|
|
define compatible keywords in g++ Used to support g++-4.6 and g++4.7
define dllexport for Visual Studio
#define MXNET_MAKE_VERSION |
( |
|
major, |
|
|
|
minor, |
|
|
|
patch |
|
) |
| ((major)*10000 + (minor)*100 + patch) |
helper for making version number
#define MXNET_PREDICT_ONLY 0 |
#define PROFILER_MESSAGE_FUNCNAME (__FUNCTION__) |
define function name as profiler message