tvm::transform::Pass Class Reference

#include <transform.h>

Inheritance diagram for tvm::transform::Pass:

Collaboration diagram for tvm::transform::Pass:

Public Member Functions
IRModule	operator() (IRModule mod) const
	Transform mod using the default PassContext in the current scope. More...
IRModule	operator() (IRModule mod, const PassContext &pass_ctx) const
	Transform mod using a functor under a given pass context. More...
	TVM_DEFINE_OBJECT_REF_METHODS (Pass, ObjectRef, PassNode)
Public Member Functions inherited from tvm::runtime::ObjectRef
	ObjectRef ()=default
	default constructor More...
	ObjectRef (ObjectPtr< Object > data)
	Constructor from existing object ptr. More...
bool	same_as (const ObjectRef &other) const
	Comparator. More...
bool	operator== (const ObjectRef &other) const
	Comparator. More...
bool	operator!= (const ObjectRef &other) const
	Comparator. More...
bool	operator< (const ObjectRef &other) const
	Comparator. More...
bool	defined () const
const Object *	get () const
const Object *	operator-> () const
bool	unique () const
int	use_count () const
template<typename ObjectType , typename = std::enable_if_t<std::is_base_of_v<Object, ObjectType>>>
const ObjectType *	as () const
	Try to downcast the internal Object to a raw pointer of a corresponding type. More...
template<typename ObjectRefType , typename = std::enable_if_t<std::is_base_of_v<ObjectRef, ObjectRefType>>>
Optional< ObjectRefType >	as () const
	Try to downcast the ObjectRef to a Optional<T> of the requested type. More...

Additional Inherited Members
Public Types inherited from tvm::runtime::ObjectRef
using	ContainerType = Object
	type indicate the container type. More...
Static Public Attributes inherited from tvm::runtime::ObjectRef
static constexpr bool	_type_is_nullable = true
Protected Member Functions inherited from tvm::runtime::ObjectRef
Object *	get_mutable () const
Static Protected Member Functions inherited from tvm::runtime::ObjectRef
template<typename T >
static T	DowncastNoCheck (ObjectRef ref)
	Internal helper function downcast a ref without check. More...
static void	FFIClearAfterMove (ObjectRef *ref)
	Clear the object ref data field without DecRef after we successfully moved the field. More...
template<typename ObjectType >
static ObjectPtr< ObjectType >	GetDataPtr (const ObjectRef &ref)
	Internal helper function get data_ as ObjectPtr of ObjectType. More...
Protected Attributes inherited from tvm::runtime::ObjectRef
ObjectPtr< Object >	data_
	Internal pointer that backs the reference. More...

Member Function Documentation

operator()() [1/2]

IRModule tvm::transform::Pass::operator()

(

IRModule

mod

)

const

Transform mod using the default PassContext in the current scope.

// If you do no longer need the input module
// it is recommended to use std::move to move your input module.
mod = pass(std::move(mod));

Parameters

mod	The module that an optimization pass runs on.

Returns

The transformed module.

operator()() [2/2]

IRModule tvm::transform::Pass::operator()	(	IRModule	mod,
		const PassContext &	pass_ctx
	)		const

Transform mod using a functor under a given pass context.

Parameters

mod	The module that an optimization pass runs on.
pass_ctx	The pass context that can provide information for the optimization.

Returns

The transformed module.

TVM_DEFINE_OBJECT_REF_METHODS()

tvm::transform::Pass::TVM_DEFINE_OBJECT_REF_METHODS	(	Pass	,
		ObjectRef	,
		PassNode
	)

---

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

• include/tvm/ir/transform.h