tvm
Loading...
Searching...
No Matches
Public Member Functions | Static Public Member Functions | Public Attributes | List of all members
tvm::transform::SequentialObj Class Reference

The SequentialObj contains a set of passes that transform Relax programs from one AST to another semantically equivalent one. More...

#include <transform.h>

Inheritance diagram for tvm::transform::SequentialObj:
Collaboration diagram for tvm::transform::SequentialObj:

Public Member Functions

 SequentialObj (PassInfo pass_info)
 
 SequentialObj (ffi::UnsafeInit tag)
 
PassInfo Info () const override
 Get the pass information/meta data.
 
IRModule operator() (IRModule mod, const PassContext &pass_ctx) const final
 Perform optimizations on a series of passes. The aforementioned typical pass manager jobs could be done by it. This function could be overloaded to focus on different metrics, i.e. performance, memory footprint, etc.
 
 TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("transform.Sequential", SequentialObj, PassObj)
 
- Public Member Functions inherited from tvm::transform::PassObj
virtual ~PassObj ()
 
IRModule operator() (IRModule mod) const
 Transform mod using the default PassContext in the current scope.
 
 TVM_FFI_DECLARE_OBJECT_INFO ("transform.Pass", PassObj, ffi::Object)
 

Static Public Member Functions

static void RegisterReflection ()
 

Public Attributes

PassInfo pass_info
 
tvm::ffi::Array< Pass > passes
 A list of passes that used to compose a sequential pass.
 

Detailed Description

The SequentialObj contains a set of passes that transform Relax programs from one AST to another semantically equivalent one.

One example of this level of pass is that the pass manager needs to correctly perform a host of optimizations with a given optimization level and disabled passes.

Constructor & Destructor Documentation

◆ SequentialObj() [1/2]

tvm::transform::SequentialObj::SequentialObj ( PassInfo  pass_info)
inlineexplicit

◆ SequentialObj() [2/2]

tvm::transform::SequentialObj::SequentialObj ( ffi::UnsafeInit  tag)
inlineexplicit

Member Function Documentation

◆ Info()

PassInfo tvm::transform::SequentialObj::Info ( ) const
inlineoverridevirtual

Get the pass information/meta data.

Implements tvm::transform::PassObj.

◆ operator()()

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

Perform optimizations on a series of passes. The aforementioned typical pass manager jobs could be done by it. This function could be overloaded to focus on different metrics, i.e. performance, memory footprint, etc.

Parameters
modThe module that these passes are applied on.
pass_ctxThe context that these passes execute on.
Returns
Return the updated module.

Implements tvm::transform::PassObj.

◆ RegisterReflection()

static void tvm::transform::SequentialObj::RegisterReflection ( )
inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::transform::SequentialObj::TVM_FFI_DECLARE_OBJECT_INFO_FINAL ( "transform.Sequential"  ,
SequentialObj  ,
PassObj   
)

Member Data Documentation

◆ pass_info

PassInfo tvm::transform::SequentialObj::pass_info

◆ passes

tvm::ffi::Array<Pass> tvm::transform::SequentialObj::passes

A list of passes that used to compose a sequential pass.


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