tvm::IRModule Class Reference

Managed reference class to IRModuleNode. More...

#include <module.h>

Inheritance diagram for tvm::IRModule:

Collaboration diagram for tvm::IRModule:

Public Types
using	ContainerType = IRModuleNode
	Declare the container type. More...

Public Member Functions
	IRModule (ffi::Map< GlobalVar, BaseFunc > functions, SourceMap map={}, DictAttrs attrs=DictAttrs(), ffi::Map< ffi::String, ffi::Array< GlobalInfo >> global_infos={})
	constructor More...
	IRModule ()
	default constructor More...
	IRModule (ObjectPtr< IRModuleNode > n)
	constructor More...
	IRModule (ffi::UnsafeInit tag)
	constructor with UnsafeInit More...
IRModuleNode *	operator-> () const
IRModule	ShallowCopyIRModule (IRModule mod)
	Create a shallow copy of an IRModule. More...
	TVM_DEFINE_OBJECT_REF_COW_METHOD (IRModuleNode)

Static Public Member Functions
static IRModule	FromExpr (const RelaxExpr &expr, const ffi::Map< GlobalVar, BaseFunc > &global_funcs={})
	As for FromExprInContext, but assuming expr is bound to 'main' and no imports. More...

Detailed Description

Managed reference class to IRModuleNode.

See also

IRModuleNode

Member Typedef Documentation

ContainerType

using tvm::IRModule::ContainerType = IRModuleNode

Declare the container type.

Constructor & Destructor Documentation

IRModule() [1/4]

tvm::IRModule::IRModule ( ffi::Map< GlobalVar, BaseFunc >  functions, SourceMap  map = {}, DictAttrs  attrs = DictAttrs(), ffi::Map< ffi::String, ffi::Array< GlobalInfo >>  global_infos = {}  )

explicit

constructor

Parameters

functions	Functions in the module.
map	The module source map.
attrs	The module meta-data attributes.
global_infos	Global infos in the module.

IRModule() [2/4]

tvm::IRModule::IRModule ( )

inline

default constructor

IRModule() [3/4]

tvm::IRModule::IRModule ( ObjectPtr< IRModuleNode >  n )

inlineexplicit

constructor

Parameters

n	The object pointer.

IRModule() [4/4]

tvm::IRModule::IRModule ( ffi::UnsafeInit  tag )

inlineexplicit

constructor with UnsafeInit

Member Function Documentation

FromExpr()

static IRModule tvm::IRModule::FromExpr ( const RelaxExpr &  expr, const ffi::Map< GlobalVar, BaseFunc > &  global_funcs = {}  )

static

As for FromExprInContext, but assuming expr is bound to 'main' and no imports.

operator->()

IRModuleNode* tvm::IRModule::operator-> ( ) const

inline

Returns

mutable pointers to the node.

ShallowCopyIRModule()

IRModule tvm::IRModule::ShallowCopyIRModule

(

IRModule

mod

)

Create a shallow copy of an IRModule.

Parameters

mod	The module to copy.

Returns

The copied module.

TVM_DEFINE_OBJECT_REF_COW_METHOD()

tvm::IRModule::TVM_DEFINE_OBJECT_REF_COW_METHOD

(

IRModuleNode

)

---

The documentation for this class was generated from the following file:

• include/tvm/ir/module.h