tvm
Loading...
Searching...
No Matches
Classes | Public Member Functions | Static Public Member Functions | List of all members
tvm::relax::BlockBuilder Class Reference

#include <block_builder.h>

Inheritance diagram for tvm::relax::BlockBuilder:
Collaboration diagram for tvm::relax::BlockBuilder:

Classes

struct  DisableOperatorSpecificNormalizationForTVMScript
 A marker struct to disable FNormalize. More...
 

Public Member Functions

 BlockBuilder (ffi::UnsafeInit tag, ffi::ObjectPtr< BlockBuilderObj > n)
 
 TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (BlockBuilder, ffi::ObjectRef, BlockBuilderObj)
 

Static Public Member Functions

static BlockBuilder Create (ffi::Optional< IRModule > ctx_mod)
 Create a BlockBuilder.
 
static BlockBuilder Create (ffi::Optional< IRModule > ctx_mod, DisableOperatorSpecificNormalizationForTVMScript tag)
 Create a BlockBuilder.
 

Constructor & Destructor Documentation

◆ BlockBuilder()

tvm::relax::BlockBuilder::BlockBuilder ( ffi::UnsafeInit  tag,
ffi::ObjectPtr< BlockBuilderObj >  n 
)
inlineexplicit

Member Function Documentation

◆ Create() [1/2]

static BlockBuilder tvm::relax::BlockBuilder::Create ( ffi::Optional< IRModule >  ctx_mod)
static

Create a BlockBuilder.

Parameters
ctx_modOptional before-transformation context module for rewriting.
Returns
The created BlockBuilder.
Note
When rewriting an existing IRModule, it is important to pass it in as ctx_mod so you can lookup the context functions for cross function call analysis.

◆ Create() [2/2]

static BlockBuilder tvm::relax::BlockBuilder::Create ( ffi::Optional< IRModule >  ctx_mod,
DisableOperatorSpecificNormalizationForTVMScript  tag 
)
static

Create a BlockBuilder.

Parameters
ctx_modOptional before-transformation context module for rewriting.
tagAn instance of DisableOperatorSpecificNormalizationForTVMScript
Returns
The created BlockBuilder.
Note
When rewriting an existing IRModule, it is important to pass it in as ctx_mod so you can lookup the context functions for cross function call analysis.

◆ TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE()

tvm::relax::BlockBuilder::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE ( BlockBuilder  ,
ffi::ObjectRef  ,
BlockBuilderObj   
)

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