tvm
Public Member Functions | Static Public Member Functions | Public Attributes | Static Public Attributes | List of all members
tvm::script::ir_builder::relax::BlockFrameNode Class Reference

The ir_builder frame for relax binding blocks. More...

#include <frame.h>

Inheritance diagram for tvm::script::ir_builder::relax::BlockFrameNode:
Collaboration diagram for tvm::script::ir_builder::relax::BlockFrameNode:

Public Member Functions

 TVM_DECLARE_FINAL_OBJECT_INFO (BlockFrameNode, RelaxFrameNode)
 
void EnterWithScope () final
 The method called when entering RAII scope. More...
 
void ExitWithScope () final
 The method called when exiting RAII scope. More...
 
- Public Member Functions inherited from tvm::script::ir_builder::relax::RelaxFrameNode
 TVM_DECLARE_BASE_OBJECT_INFO (RelaxFrameNode, IRBuilderFrameNode)
 
- Public Member Functions inherited from tvm::script::ir_builder::IRBuilderFrameNode
 TVM_DECLARE_BASE_OBJECT_INFO (IRBuilderFrameNode, runtime::Object)
 
virtual ~IRBuilderFrameNode ()=default
 Default destructor. More...
 
void AddCallback (ffi::TypedFunction< void()> callback)
 Add a callback method invoked when exiting the RAII scope. More...
 

Static Public Member Functions

static void RegisterReflection ()
 
- Static Public Member Functions inherited from tvm::script::ir_builder::relax::RelaxFrameNode
static void RegisterReflection ()
 
- Static Public Member Functions inherited from tvm::script::ir_builder::IRBuilderFrameNode
static void RegisterReflection ()
 

Public Attributes

bool is_dataflow
 The flag that indicates whether the block is a dataflow block. More...
 
Array< tvm::relax::Varemitted_vars
 The variables emitted in this block. More...
 
bool block_ended
 A boolean indicating if the dataflow block is ended of construction. If it is true, any new binding trying to be emitted into this block will cause an error. More...
 
Array< tvm::relax::Varoutput_vars
 The output vars of the dataflow block. More...
 
- Public Attributes inherited from tvm::script::ir_builder::IRBuilderFrameNode
std::vector< ffi::TypedFunction< void()> > callbacks
 A list of callbacks used when exiting the frame. More...
 

Static Public Attributes

static constexpr const char * _type_key = "script.ir_builder.relax.BlockFrame"
 
- Static Public Attributes inherited from tvm::script::ir_builder::relax::RelaxFrameNode
static constexpr const char * _type_key = "script.ir_builder.relax.RelaxFrame"
 
- Static Public Attributes inherited from tvm::script::ir_builder::IRBuilderFrameNode
static constexpr const char * _type_key = "script.ir_builder.IRBuilderFrame"
 

Detailed Description

The ir_builder frame for relax binding blocks.

Member Function Documentation

◆ EnterWithScope()

void tvm::script::ir_builder::relax::BlockFrameNode::EnterWithScope ( )
finalvirtual

The method called when entering RAII scope.

See also
tvm::support::With

Reimplemented from tvm::script::ir_builder::IRBuilderFrameNode.

◆ ExitWithScope()

void tvm::script::ir_builder::relax::BlockFrameNode::ExitWithScope ( )
finalvirtual

The method called when exiting RAII scope.

See also
tvm::support::With

Reimplemented from tvm::script::ir_builder::IRBuilderFrameNode.

◆ RegisterReflection()

static void tvm::script::ir_builder::relax::BlockFrameNode::RegisterReflection ( )
inlinestatic

◆ TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::script::ir_builder::relax::BlockFrameNode::TVM_DECLARE_FINAL_OBJECT_INFO ( BlockFrameNode  ,
RelaxFrameNode   
)

Member Data Documentation

◆ _type_key

constexpr const char* tvm::script::ir_builder::relax::BlockFrameNode::_type_key = "script.ir_builder.relax.BlockFrame"
staticconstexpr

◆ block_ended

bool tvm::script::ir_builder::relax::BlockFrameNode::block_ended

A boolean indicating if the dataflow block is ended of construction. If it is true, any new binding trying to be emitted into this block will cause an error.

Note
Only used for a dataflow block.

◆ emitted_vars

Array<tvm::relax::Var> tvm::script::ir_builder::relax::BlockFrameNode::emitted_vars

The variables emitted in this block.

◆ is_dataflow

bool tvm::script::ir_builder::relax::BlockFrameNode::is_dataflow

The flag that indicates whether the block is a dataflow block.

◆ output_vars

Array<tvm::relax::Var> tvm::script::ir_builder::relax::BlockFrameNode::output_vars

The output vars of the dataflow block.

Note
Only used for a dataflow block.

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