mxnet
Classes | Namespaces | Macros | Typedefs
operator.h File Reference
#include <dmlc/base.h>
#include <dmlc/json.h>
#include <dmlc/logging.h>
#include <dmlc/registry.h>
#include <nnvm/node.h>
#include <vector>
#include <map>
#include <string>
#include <utility>
#include "./base.h"
#include "./resource.h"
#include "./op_attr_types.h"
Include dependency graph for operator.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  mxnet::Operator
 Operator interface. Operator defines basic operation unit of optimized computation graph in mxnet. This interface relies on pre-allocated memory in TBlob, the caller need to set the memory region in TBlob correctly before calling Forward and Backward. More...
 
class  mxnet::OperatorProperty
 OperatorProperty is a object that stores all information about Operator. It also contains method to generate context(device) specific operators. More...
 
struct  mxnet::OperatorPropertyReg
 Registry entry for OperatorProperty factory functions. More...
 

Namespaces

 mxnet
 namespace of mxnet
 

Macros

#define MXNET_REGISTER_OP_PROPERTY(name, OperatorPropertyType)
 Macro to register OperatorProperty. More...
 

Typedefs

typedef std::function< OperatorProperty *()> mxnet::OperatorPropertyFactory
 typedef the factory function of operator property More...
 

Macro Definition Documentation

#define MXNET_REGISTER_OP_PROPERTY (   name,
  OperatorPropertyType 
)
Value:
DMLC_REGISTRY_REGISTER(::mxnet::OperatorPropertyReg, OperatorPropertyReg, name) \
.set_body([]() { return new OperatorPropertyType(); }) \
.set_return_type("NDArray-or-Symbol") \
.check_name()
Registry entry for OperatorProperty factory functions.
Definition: operator.h:452

Macro to register OperatorProperty.

1 // example of registering a fully connected operator
2 REGISTER_OP_PROPERTY(FullyConnected, FullyConnectedOpProp)
3 .describe("Fully connected layer");