tvm
Public Member Functions | List of all members
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)
 

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));
tvm::PrimExpr mod(const tvm::PrimExpr &a, const tvm::PrimExpr &b)
Definition: broadcast.h:306
Parameters
modThe 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
modThe module that an optimization pass runs on.
pass_ctxThe 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: