A frame that represents the PrimFunc containing TIR statements. More...

#include <frame.h>

Inheritance diagram for tvm::script::ir_builder::tir::PrimFuncFrameNode:

Collaboration diagram for tvm::script::ir_builder::tir::PrimFuncFrameNode:

Public Member Functions
	TVM_DECLARE_FINAL_OBJECT_INFO (PrimFuncFrameNode, TIRFrameNode)

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

Public Member Functions inherited from tvm::script::ir_builder::tir::TIRFrameNode
	TVM_DECLARE_BASE_OBJECT_INFO (TIRFrameNode, 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...

virtual void	EnterWithScope ()
	The method called when entering RAII scope. 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::tir::TIRFrameNode
static void	RegisterReflection ()

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

Public Attributes
Optional< String >	name
	The name of the block. More...

Array< tvm::tir::Var >	args
	Function parameters. More...

bool	is_private
	Whether the PrimFunc is annotated as private. More...

Optional< Type >	ret_type
	The return type of the function. More...

Map< tvm::tir::Var, tvm::tir::Buffer >	buffer_map
	Maps some parameters to specific Buffer data structures. More...

Map< String, Any >	attrs
	Additional attributes storing the meta-data. More...

Map< tvm::tir::Var, tvm::tir::IterVar >	env_threads
	The variable map bound to thread env. More...

Array< tvm::tir::Buffer >	root_alloc_buffers
	The buffer allocated in root block. More...

Public Attributes inherited from tvm::script::ir_builder::tir::TIRFrameNode
Array< tvm::tir::Stmt >	stmts
	The Stmt within in this frame. 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.tir.PrimFuncFrame"

Static Public Attributes inherited from tvm::script::ir_builder::tir::TIRFrameNode
static constexpr const char *	_type_key = "script.ir_builder.tir.TIRFrame"

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

Detailed Description

A frame that represents the PrimFunc containing TIR statements.

See also: PrimFuncFrame

Member Function Documentation

◆ ExitWithScope()

void tvm::script::ir_builder::tir::PrimFuncFrameNode::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::tir::PrimFuncFrameNode::RegisterReflection ( )

inlinestatic

◆ TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::script::ir_builder::tir::PrimFuncFrameNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	PrimFuncFrameNode	,
		TIRFrameNode
	)

Member Data Documentation

◆ _type_key

constexpr const char* tvm::script::ir_builder::tir::PrimFuncFrameNode::_type_key = "script.ir_builder.tir.PrimFuncFrame"

staticconstexpr

◆ args

Array<tvm::tir::Var> tvm::script::ir_builder::tir::PrimFuncFrameNode::args

Function parameters.

◆ attrs

Map<String, Any> tvm::script::ir_builder::tir::PrimFuncFrameNode::attrs

Additional attributes storing the meta-data.

◆ buffer_map

Map<tvm::tir::Var, tvm::tir::Buffer> tvm::script::ir_builder::tir::PrimFuncFrameNode::buffer_map

Maps some parameters to specific Buffer data structures.

◆ env_threads

Map<tvm::tir::Var, tvm::tir::IterVar> tvm::script::ir_builder::tir::PrimFuncFrameNode::env_threads

The variable map bound to thread env.

◆ is_private

bool tvm::script::ir_builder::tir::PrimFuncFrameNode::is_private

Whether the PrimFunc is annotated as private.

◆ name

Optional<String> tvm::script::ir_builder::tir::PrimFuncFrameNode::name

The name of the block.

◆ ret_type

Optional<Type> tvm::script::ir_builder::tir::PrimFuncFrameNode::ret_type

The return type of the function.

◆ root_alloc_buffers

Array<tvm::tir::Buffer> tvm::script::ir_builder::tir::PrimFuncFrameNode::root_alloc_buffers

The buffer allocated in root block.

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

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

Public Member Functions

Static Public Member Functions

Public Attributes

Static Public Attributes