25 #ifndef NNVM_COMPILER_PACKED_FUNC_EXT_H_ 26 #define NNVM_COMPILER_PACKED_FUNC_EXT_H_ 28 #include <tvm/runtime/packed_func.h> 29 #include <tvm/runtime/registry.h> 34 #include <unordered_map> 39 using tvm::runtime::PackedFunc;
41 using AttrDict = std::unordered_map<std::string, std::string>;
50 const PackedFunc* pf = tvm::runtime::Registry::Get(name);
51 CHECK(pf !=
nullptr) <<
"Cannot find function " << name <<
" in registry";
62 struct extension_type_info<
nnvm::Symbol> {
63 static const int code = 16;
67 struct extension_type_info<
nnvm::Graph> {
68 static const int code = 17;
73 static const int code = 18;
78 #endif // NNVM_COMPILER_PACKED_FUNC_EXT_H_
Definition: packed_func_ext.h:58
Configuation of nnvm as well as basic data structure.
Symbolic graph construction API.
const PackedFunc & GetPackedFunc(const std::string &name)
Get PackedFunction from global registry and report error if it does not exist.
Definition: packed_func_ext.h:49
std::unordered_map< std::string, std::string > AttrDict
Definition: packed_func_ext.h:41