mxnet
|
Operator information structor. More...
#include <dmlc/parameter.h>
#include <string>
#include <vector>
#include <utility>
#include <typeinfo>
#include <limits>
#include <functional>
#include "base.h"
#include "c_api.h"
Go to the source code of this file.
Classes | |
class | nnvm::OpMap< ValueType > |
A map data structure that takes Op* as key and returns ValueType. More... | |
class | nnvm::Op |
Operator structure. More... | |
class | nnvm::OpMap< ValueType > |
A map data structure that takes Op* as key and returns ValueType. More... | |
class | nnvm::OpGroup |
auxiliary data structure used to set attributes to a group of operators More... | |
Namespaces | |
nnvm | |
Macros | |
#define | NNVM_REGISTER_VAR_DEF(OpName) static DMLC_ATTRIBUTE_UNUSED ::nnvm::Op & __make_ ## NnvmOp ## _ ## OpName |
#define | NNVM_REGISTER_GVAR_DEF(TagName) static DMLC_ATTRIBUTE_UNUSED ::nnvm::OpGroup __make_ ## NnvmOpGroup ## _ ## TagName |
#define | NNVM_REGISTER_OP(OpName) |
Register a new operator, or set attribute of the corresponding op. More... | |
#define | NNVM_REGISTER_OP_GROUP(GroupName) |
Register attribute to a group of operators. These attributes will be registered to Op that include the group. More... | |
Operator information structor.
#define NNVM_REGISTER_GVAR_DEF | ( | TagName | ) | static DMLC_ATTRIBUTE_UNUSED ::nnvm::OpGroup __make_ ## NnvmOpGroup ## _ ## TagName |
#define NNVM_REGISTER_OP | ( | OpName | ) |
Register a new operator, or set attribute of the corresponding op.
OpName | The name of registry |
#define NNVM_REGISTER_OP_GROUP | ( | GroupName | ) |
Register attribute to a group of operators. These attributes will be registered to Op that include the group.
GroupName | The name of the group. |
#define NNVM_REGISTER_VAR_DEF | ( | OpName | ) | static DMLC_ATTRIBUTE_UNUSED ::nnvm::Op & __make_ ## NnvmOp ## _ ## OpName |