mxnet
Public Member Functions | Static Public Member Functions | List of all members
mxnet::op::SimpleOpRegistry Class Reference

registry for TBlob functions More...

#include <operator_util.h>

Collaboration diagram for mxnet::op::SimpleOpRegistry:
Collaboration graph

Public Member Functions

SimpleOpRegEntry__REGISTER_OR_FIND__ (char const *name)
 Internal function to register a name function under name. More...
 

Static Public Member Functions

static const SimpleOpRegEntryFind (const std::string &name)
 Find the entry with corresponding name. More...
 
static SimpleOpRegistryGet ()
 

Detailed Description

registry for TBlob functions

Member Function Documentation

SimpleOpRegEntry& mxnet::op::SimpleOpRegistry::__REGISTER_OR_FIND__ ( char const *  name)

Internal function to register a name function under name.

Parameters
namename of the function
Returns
ref to the registered entry, used to set properties
static const SimpleOpRegEntry* mxnet::op::SimpleOpRegistry::Find ( const std::string &  name)
inlinestatic

Find the entry with corresponding name.

Parameters
namename of the function
Returns
the corresponding function, can be NULL
static SimpleOpRegistry* mxnet::op::SimpleOpRegistry::Get ( )
static
Returns
global singleton of the registry

The documentation for this class was generated from the following file: