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

#include <frame.h>

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

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

Public Member Functions
	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("script.ir_builder.relax.BindingBlockFrame", BindingBlockFrameObj, RelaxFrameObj)

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::RelaxFrameObj
	TVM_FFI_DECLARE_OBJECT_INFO ("script.ir_builder.relax.RelaxFrame", RelaxFrameObj, IRBuilderFrameObj)

Public Member Functions inherited from tvm::script::ir_builder::IRBuilderFrameObj
	IRBuilderFrameObj ()
	Capture the source-call context when this frame is constructed.

	TVM_FFI_DECLARE_OBJECT_INFO ("script.ir_builder.IRBuilderFrame", IRBuilderFrameObj, ffi::Object)

virtual	~IRBuilderFrameObj ()=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::RelaxFrameObj
static void	RegisterReflection ()

Static Public Member Functions inherited from tvm::script::ir_builder::IRBuilderFrameObj
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.

ffi::Map< tvm::Var, Location >	binding_locs
	Statement ranges for explicitly emitted bindings.

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

Location	loc = UnknownLoc()
	Source context retained until this frame constructs its result.

Additional Inherited Members
Static Public Attributes inherited from tvm::script::ir_builder::IRBuilderFrameObj
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::BindingBlockFrameObj::EnterWithScope ( )

finalvirtual

The method called when entering RAII scope.

See also: tvm::support::With

Reimplemented from tvm::script::ir_builder::IRBuilderFrameObj.

◆ ExitWithScope()

void tvm::script::ir_builder::relax::BindingBlockFrameObj::ExitWithScope ( )

finalvirtual

The method called when exiting RAII scope.

See also: tvm::support::With

Reimplemented from tvm::script::ir_builder::IRBuilderFrameObj.

◆ RegisterReflection()

static void tvm::script::ir_builder::relax::BindingBlockFrameObj::RegisterReflection ( )

inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::script::ir_builder::relax::BindingBlockFrameObj::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"script.ir_builder.relax.BindingBlockFrame"	,
		BindingBlockFrameObj	,
		RelaxFrameObj
	)

Member Data Documentation

◆ binding_locs

ffi::Map<tvm::Var, Location> tvm::script::ir_builder::relax::BindingBlockFrameObj::binding_locs

Statement ranges for explicitly emitted bindings.

◆ block_ended

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

The variables emitted in this block.

◆ is_dataflow

bool tvm::script::ir_builder::relax::BindingBlockFrameObj::is_dataflow

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

◆ output_vars

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

Public Member Functions

Static Public Member Functions

Public Attributes

Additional Inherited Members