tvm::script::ir_builder::tir::PrimFuncFrameNode Class Reference

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_FFI_DECLARE_OBJECT_INFO_FINAL ("script.ir_builder.tir.PrimFuncFrame", 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_FFI_DECLARE_OBJECT_INFO ("script.ir_builder.tir.TIRFrame", TIRFrameNode, 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...
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
ffi::Optional< ffi::String >	name
	The name of the block. More...
ffi::Array< tvm::tir::Var >	args
	Function parameters. More...
bool	is_private
	Whether the PrimFunc is annotated as private. More...
ffi::Optional< Type >	ret_type
	The return type of the function. More...
ffi::Map< tvm::tir::Var, tvm::tir::Buffer >	buffer_map
	Maps some parameters to specific Buffer data structures. More...
ffi::Map< ffi::String, Any >	attrs
	Additional attributes storing the meta-data. More...
ffi::Map< tvm::tir::Var, tvm::tir::IterVar >	env_threads
	The variable map bound to thread env. More...
ffi::Array< tvm::tir::Buffer >	root_alloc_buffers
	The buffer allocated in root block. More...
Public Attributes inherited from tvm::script::ir_builder::tir::TIRFrameNode
ffi::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...

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

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

tvm::script::ir_builder::tir::PrimFuncFrameNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"script.ir_builder.tir.PrimFuncFrame"	,
		PrimFuncFrameNode	,
		TIRFrameNode
	)

Member Data Documentation

args

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

Function parameters.

attrs

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

Additional attributes storing the meta-data.

buffer_map

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

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

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

The name of the block.

ret_type

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

The return type of the function.

root_alloc_buffers

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