api_registry.h File Reference

This file contains utilities related to the MXNet's global function registry. More...

#include <string>
#include <utility>
#include "runtime/registry.h"

Include dependency graph for api_registry.h:

Go to the source code of this file.

Namespaces
	mxnet
	namespace of mxnet

Macros
#define	MXNET_REGISTER_API(OpName)   MXNET_REGISTER_GLOBAL(OpName)
	Register an API function globally. It simply redirects to MXNET_REGISTER_GLOBAL. More...

Detailed Description

This file contains utilities related to the MXNet's global function registry.

Macro Definition Documentation

#define MXNET_REGISTER_API

(

OpName

)

MXNET_REGISTER_GLOBAL(OpName)

Register an API function globally. It simply redirects to MXNET_REGISTER_GLOBAL.

MXNET_REGISTER_API(MyPrint)
.set_body([](MXNetArgs args, MXNetRetValue* rv) {
  // my code.
});