module.h File Reference

IRModule that holds the functions and type definitions. More...

#include <tvm/ir/adt.h>
#include <tvm/ir/expr.h>
#include <tvm/ir/function.h>
#include <tvm/ir/type.h>
#include <tvm/parser/source_map.h>
#include <tvm/runtime/container/array.h>
#include <tvm/runtime/container/map.h>
#include <tvm/runtime/container/string.h>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>

Include dependency graph for module.h:

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

Go to the source code of this file.

Classes
class	tvm::IRModuleNode
	IRModule that holds functions and type definitions. More...
class	tvm::IRModule
	Managed reference class to IRModuleNode. More...

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::attr
	Generic attribute names that can be attached to any function.

Functions
String	tvm::PrettyPrint (const ObjectRef &node)
	Pretty print a node for debug purposes. More...
String	tvm::AsText (const ObjectRef &node, bool show_meta_data=true, runtime::TypedPackedFunc< String(ObjectRef)> annotate=nullptr)
	Render the node as a string in the text format. More...

Variables
constexpr const char *	tvm::attr::kModuleName = "mod_name"
	Name of the module. More...
constexpr const char *	tvm::attr::kExecutor = "executor"
	Executor targeted by the module. More...
constexpr const char *	tvm::attr::kRuntime = "runtime"
	Runtime target of the module. More...
constexpr const char *	tvm::attr::kWorkspaceMemoryPools = "workspace_memory_pools"
	workspace memory pools of the module More...
constexpr const char *	tvm::attr::kConstantMemoryPools = "constant_memory_pools"
	constant memory pools of the module More...
constexpr const char *	tvm::attr::kConstants = "constants"
constexpr const char *	tvm::attr::kExternalMods = "external_mods"
	All the runtime::Modules accumulated during compilation by external codegen. These modules must be either directly linked or captured in the final compilation artifact. More...
constexpr const char *	tvm::attr::kConstNameToConstant = "const_name_to_constant"
	All the named runtime::NDArrays accumulated during compilation by external codegen. Generally the associated runtime::Module will indicate it requires bindings for these names, and during module initialization these bindings will be recovered from a ConstLoaderModule. See also kConstantsArray above, which is the analog for PrimFuncs. More...

Detailed Description

IRModule that holds the functions and type definitions.