tvm
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

 TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS (BlockBuilder, ObjectRef, BlockBuilderNode)
 

Static Public Member Functions

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

Member Function Documentation

◆ Create() [1/2]

static BlockBuilder tvm::relax::BlockBuilder::Create ( 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 ( 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_DEFINE_MUTABLE_OBJECT_REF_METHODS()

tvm::relax::BlockBuilder::TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS ( BlockBuilder  ,
ObjectRef  ,
BlockBuilderNode   
)

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