mxnet
Classes | Namespaces | Macros
registry.h File Reference
#include <string>
#include <vector>
#include "packed_func.h"
Include dependency graph for registry.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  mxnet::runtime::Registry
 Registry for global function. More...
 

Namespaces

 mxnet
 namespace of mxnet
 
 mxnet::runtime
 

Macros

#define MXNET_ATTRIBUTE_UNUSED
 helper macro to supress unused warning More...
 
#define MXNET_STR_CONCAT_(__x, __y)   __x##__y
 
#define MXNET_STR_CONCAT(__x, __y)   MXNET_STR_CONCAT_(__x, __y)
 
#define MXNET_FUNC_REG_VAR_DEF   static MXNET_ATTRIBUTE_UNUSED ::mxnet::runtime::Registry& __mk_ ## MXNET
 
#define MXNET_REGISTER_GLOBAL(OpName)
 Register a function globally. More...
 

Macro Definition Documentation

#define MXNET_ATTRIBUTE_UNUSED

helper macro to supress unused warning

#define MXNET_FUNC_REG_VAR_DEF   static MXNET_ATTRIBUTE_UNUSED ::mxnet::runtime::Registry& __mk_ ## MXNET
#define MXNET_REGISTER_GLOBAL (   OpName)
Value:
#define MXNET_FUNC_REG_VAR_DEF
Definition: registry.h:297
static MXNET_DLL Registry & Register(const std::string &name, bool override=false)
Register a function with given name.
#define MXNET_STR_CONCAT(__x, __y)
Definition: registry.h:295

Register a function globally.

1 TVM_REGISTER_GLOBAL("MyPrint")
2 .set_body([](TVMArgs args, TVMRetValue* rv) {
3 });
#define MXNET_STR_CONCAT (   __x,
  __y 
)    MXNET_STR_CONCAT_(__x, __y)
#define MXNET_STR_CONCAT_ (   __x,
  __y 
)    __x##__y