packed_func_ext.h File Reference

Extension to enable packed functionn for nnvm types. More...

#include <tvm/runtime/packed_func.h>
#include <tvm/runtime/registry.h>
#include <nnvm/graph.h>
#include <nnvm/symbolic.h>
#include <string>
#include <vector>
#include <unordered_map>

Include dependency graph for packed_func_ext.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	tvm::runtime::extension_type_info< nnvm::Symbol >
struct	tvm::runtime::extension_type_info< nnvm::Graph >
struct	tvm::runtime::extension_type_info< nnvm::compiler::AttrDict >

Namespaces
	nnvm
	nnvm::compiler
	tvm
	tvm::runtime

Typedefs
using	nnvm::compiler::AttrDict = std::unordered_map< std::string, std::string >

Functions
const PackedFunc &	nnvm::compiler::GetPackedFunc (const std::string &name)
	Get PackedFunction from global registry and report error if it does not exist. More...

Detailed Description

Extension to enable packed functionn for nnvm types.

Copyright (c) 2017 by Contributors