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_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (Pass, ObjectRef, PassNode)

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_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE()

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

---

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

• include/tvm/ir/transform.h