operator.h File Reference

Operator interface of mxnet. More...

#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...

Detailed Description

Operator interface of mxnet.

Author

Naiyan Wang

Macro Definition Documentation

MXNET_REGISTER_OP_PROPERTY

#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()

Macro to register OperatorProperty.

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