traced_object_functor.h File Reference

#include <tvm/node/node.h>
#include <tvm/runtime/logging.h>
#include <tvm/runtime/packed_func.h>
#include <tvm/script/printer/traced_object.h>
#include <string>
#include <type_traits>
#include <unordered_map>
#include <utility>
#include <vector>

Include dependency graph for traced_object_functor.h:

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

Go to the source code of this file.

Classes
class	tvm::script::printer::TracedObjectFunctor< R, Args >
	Dynamic dispatch functor based on TracedObject. More...

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::script
	tvm::script::printer

Typedefs
using	tvm::script::printer::DispatchTable = std::unordered_map< std::string, std::vector< runtime::PackedFunc > >

Functions
const runtime::PackedFunc &	tvm::script::printer::GetDispatchFunction (const DispatchTable &dispatch_table, const String &token, uint32_t type_index)
	Get function from dispatch table. More...
void	tvm::script::printer::SetDispatchFunction (DispatchTable *dispatch_table, const String &token, uint32_t type_index, runtime::PackedFunc f)
	Set function in dispatch table. More...
void	tvm::script::printer::RemoveDispatchFunction (DispatchTable *dispatch_table, const String &token, uint32_t type_index)
	Remove function from dispatch table. More...

Variables
constexpr const char *	tvm::script::printer::kDefaultDispatchToken = ""