registry for TBlob functions
More...
#include <operator_util.h>
registry for TBlob functions
SimpleOpRegEntry& mxnet::op::SimpleOpRegistry::__REGISTER_OR_FIND__ |
( |
char const * |
name | ) |
|
Internal function to register a name function under name.
- Parameters
-
- 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
-
- Returns
- the corresponding function, can be NULL
- Returns
- global singleton of the registry
The documentation for this class was generated from the following file: