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_FFI_DECLARE_OBJECT_INFO_FINAL ("script.ir_builder.relax.BlockFrame", 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_FFI_DECLARE_OBJECT_INFO ("script.ir_builder.relax.RelaxFrame", RelaxFrameNode, IRBuilderFrameNode)

Public Member Functions inherited from tvm::script::ir_builder::IRBuilderFrameNode
	TVM_FFI_DECLARE_OBJECT_INFO ("script.ir_builder.IRBuilderFrame", 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...

ffi::Array< tvm::relax::Var >	emitted_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...

ffi::Array< tvm::relax::Var >	output_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...

Additional Inherited Members
Static Public Attributes inherited from tvm::script::ir_builder::IRBuilderFrameNode
static constexpr const bool	_type_mutable = true

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_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::script::ir_builder::relax::BlockFrameNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"script.ir_builder.relax.BlockFrame"	,
		BlockFrameNode	,
		RelaxFrameNode
	)

Member Data Documentation

◆ 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

ffi::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

ffi::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:

include/tvm/script/ir_builder/relax/frame.h

Public Member Functions

Static Public Member Functions

Public Attributes

Additional Inherited Members