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

#include <frame.h>

Inheritance diagram for tvm::script::ir_builder::relax::BindingBlockFrameNode:

Collaboration diagram for tvm::script::ir_builder::relax::BindingBlockFrameNode:

Public Member Functions
	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("script.ir_builder.relax.BindingBlockFrame", BindingBlockFrameNode, RelaxFrameNode)

void	EnterWithScope () final
	The method called when entering RAII scope.

void	ExitWithScope () final
	The method called when exiting RAII scope.

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, ffi::Object)

virtual	~IRBuilderFrameNode ()=default
	Default destructor.

void	AddCallback (ffi::TypedFunction< void()> callback)
	Add a callback method invoked when exiting the RAII scope.

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.

ffi::Array< tvm::Var >	emitted_vars
	The variables emitted in this block.

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.

ffi::Array< tvm::Var >	output_vars
	The output vars of the dataflow block.

Public Attributes inherited from tvm::script::ir_builder::IRBuilderFrameNode
std::vector< ffi::TypedFunction< void()> >	callbacks
	A list of callbacks used when exiting the frame.

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::BindingBlockFrameNode::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::BindingBlockFrameNode::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::BindingBlockFrameNode::RegisterReflection ( )

inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::script::ir_builder::relax::BindingBlockFrameNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"script.ir_builder.relax.BindingBlockFrame"	,
		BindingBlockFrameNode	,
		RelaxFrameNode
	)

Member Data Documentation

◆ block_ended

bool tvm::script::ir_builder::relax::BindingBlockFrameNode::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::Var> tvm::script::ir_builder::relax::BindingBlockFrameNode::emitted_vars

The variables emitted in this block.

◆ is_dataflow

bool tvm::script::ir_builder::relax::BindingBlockFrameNode::is_dataflow

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

◆ output_vars

ffi::Array<tvm::Var> tvm::script::ir_builder::relax::BindingBlockFrameNode::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/relax/script/builder/frame.h

Public Member Functions

Static Public Member Functions

Public Attributes

Additional Inherited Members