mxnet
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
Classes | Namespaces | Macros | Typedefs | Enumerations
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 <vector>
#include <map>
#include <string>
#include <utility>
#include "./base.h"
#include "./resource.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

struct  mxnet::OpContext
 All the possible information needed by Operator.Forward and Backward This is the superset of RunContext. We use this data structure to bookkeep everything needed by Forward and Backward. More...
 
class  mxnet::Operator
 Operator interface. Operator defins 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...
 

Enumerations

enum  mxnet::OpReqType { mxnet::kNullOp, mxnet::kWriteTo, mxnet::kWriteInplace, mxnet::kAddTo }
 operation request type to Forward and Backward More...
 

Detailed Description

Operator interface of mxnet.

Copyright (c) 2015 by Contributors

Author
Naiyan Wang

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("Symbol") \
.check_name()
Registry entry for OperatorProperty factory functions.
Definition: operator.h:487

Macro to register OperatorProperty.

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