tvm
Public Member Functions | Static Public Member Functions | Public Attributes | Static Public Attributes | List of all members
tvm::script::ir_builder::tir::BlockFrameNode Class Reference

A frame that represents the block. More...

#include <frame.h>

Inheritance diagram for tvm::script::ir_builder::tir::BlockFrameNode:
Collaboration diagram for tvm::script::ir_builder::tir::BlockFrameNode:

Public Member Functions

 TVM_DECLARE_FINAL_OBJECT_INFO (BlockFrameNode, 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

String name
 The name of the block. More...
 
Array< tvm::tir::IterVariter_vars
 The variables of the block. More...
 
Optional< Array< tvm::tir::BufferRegion > > reads
 The read buffer regions of the block. More...
 
Optional< Array< tvm::tir::BufferRegion > > writes
 The write buffer regions of the block. More...
 
Optional< tvm::tir::Stmtinit
 The init statement of the bolck. More...
 
Array< tvm::tir::Bufferalloc_buffers
 The buffer allocated in the block. More...
 
Array< tvm::tir::MatchBufferRegionmatch_buffers
 The match buffer regions. More...
 
Optional< Map< String, Any > > annotations
 The annotation of the block. More...
 
Array< PrimExpriter_values
 The corresponding values of the iter vars. More...
 
Optional< PrimExprpredicate
 The predicate of the block realization, the block will only be executed when the predicate is true. More...
 
bool no_realize
 The flag whether to construct BlockRealize or Block. More...
 
- Public Attributes inherited from tvm::script::ir_builder::tir::TIRFrameNode
Array< tvm::tir::Stmtstmts
 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.BlockFrame"
 
- 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 block.

See also
BlockFrame

Member Function Documentation

◆ ExitWithScope()

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

◆ TVM_DECLARE_FINAL_OBJECT_INFO()

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

Member Data Documentation

◆ _type_key

constexpr const char* tvm::script::ir_builder::tir::BlockFrameNode::_type_key = "script.ir_builder.tir.BlockFrame"
staticconstexpr

◆ alloc_buffers

Array<tvm::tir::Buffer> tvm::script::ir_builder::tir::BlockFrameNode::alloc_buffers

The buffer allocated in the block.

◆ annotations

Optional<Map<String, Any> > tvm::script::ir_builder::tir::BlockFrameNode::annotations

The annotation of the block.

◆ init

Optional<tvm::tir::Stmt> tvm::script::ir_builder::tir::BlockFrameNode::init

The init statement of the bolck.

◆ iter_values

Array<PrimExpr> tvm::script::ir_builder::tir::BlockFrameNode::iter_values

The corresponding values of the iter vars.

◆ iter_vars

Array<tvm::tir::IterVar> tvm::script::ir_builder::tir::BlockFrameNode::iter_vars

The variables of the block.

◆ match_buffers

Array<tvm::tir::MatchBufferRegion> tvm::script::ir_builder::tir::BlockFrameNode::match_buffers

The match buffer regions.

◆ name

String tvm::script::ir_builder::tir::BlockFrameNode::name

The name of the block.

◆ no_realize

bool tvm::script::ir_builder::tir::BlockFrameNode::no_realize

The flag whether to construct BlockRealize or Block.

◆ predicate

Optional<PrimExpr> tvm::script::ir_builder::tir::BlockFrameNode::predicate

The predicate of the block realization, the block will only be executed when the predicate is true.

◆ reads

Optional<Array<tvm::tir::BufferRegion> > tvm::script::ir_builder::tir::BlockFrameNode::reads

The read buffer regions of the block.

◆ writes

Optional<Array<tvm::tir::BufferRegion> > tvm::script::ir_builder::tir::BlockFrameNode::writes

The write buffer regions of the block.


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