mxnet
|
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>
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... | |
Extension to enable packed functionn for nnvm types.
Copyright (c) 2017 by Contributors