28 строки
636 B
C
28 строки
636 B
C
/*!
|
|
* Copyright (c) 2017 by Contributors
|
|
* \file api_registry.h
|
|
* \brief This files include necessary headers to
|
|
* be used to register an global API function.
|
|
*/
|
|
#ifndef TVM_API_REGISTRY_H_
|
|
#define TVM_API_REGISTRY_H_
|
|
|
|
#include "./base.h"
|
|
#include "./packed_func_ext.h"
|
|
#include "./runtime/registry.h"
|
|
|
|
/*!
|
|
* \brief Register an API function globally.
|
|
* It simply redirects to TVM_REGISTER_GLOBAL
|
|
*
|
|
* \code
|
|
* TVM_REGISTER_API(MyPrint)
|
|
* .set_body([](TVMArgs args, TVMRetValue* rv) {
|
|
* // my code.
|
|
* });
|
|
* \endcode
|
|
*/
|
|
#define TVM_REGISTER_API(OpName) TVM_REGISTER_GLOBAL(OpName)
|
|
|
|
#endif // TVM_API_REGISTRY_H_
|