tvm
|
A frame that represents the block. More...
#include <frame.h>
Public Member Functions | |
void | VisitAttrs (tvm::AttrVisitor *v) |
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 | |
void | VisitAttrs (tvm::AttrVisitor *v) |
TVM_DECLARE_BASE_OBJECT_INFO (TIRFrameNode, IRBuilderFrameNode) | |
Public Member Functions inherited from tvm::script::ir_builder::IRBuilderFrameNode | |
void | VisitAttrs (tvm::AttrVisitor *v) |
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 (runtime::TypedPackedFunc< void()> callback) |
Add a callback method invoked when exiting the RAII scope. More... | |
Public Member Functions inherited from tvm::runtime::Object | |
uint32_t | type_index () const |
std::string | GetTypeKey () const |
size_t | GetTypeKeyHash () const |
template<typename TargetType > | |
bool | IsInstance () const |
bool | unique () const |
Object () | |
Object (const Object &other) | |
Object (Object &&other) | |
Object & | operator= (const Object &other) |
Object & | operator= (Object &&other) |
Public Attributes | |
String | name |
The name of the block. More... | |
Array< tvm::tir::IterVar > | iter_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::Stmt > | init |
The init statement of the bolck. More... | |
Array< tvm::tir::Buffer > | alloc_buffers |
The buffer allocated in the block. More... | |
Array< tvm::tir::MatchBufferRegion > | match_buffers |
The match buffer regions. More... | |
Optional< Map< String, ObjectRef > > | annotations |
The annotation of the block. More... | |
Array< PrimExpr > | iter_values |
The corresponding values of the iter vars. More... | |
Optional< PrimExpr > | predicate |
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::Stmt > | stmts |
The Stmt within in this frame. More... | |
Public Attributes inherited from tvm::script::ir_builder::IRBuilderFrameNode | |
std::vector< runtime::TypedPackedFunc< 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" |
Static Public Attributes inherited from tvm::runtime::Object | |
static constexpr const char * | _type_key = "runtime.Object" |
static constexpr bool | _type_final = false |
static constexpr uint32_t | _type_child_slots = 0 |
static constexpr bool | _type_child_slots_can_overflow = true |
static constexpr bool | _type_has_method_visit_attrs = true |
static constexpr bool | _type_has_method_sequal_reduce = false |
static constexpr bool | _type_has_method_shash_reduce = false |
static constexpr uint32_t | _type_index = TypeIndex::kDynamic |
Additional Inherited Members | |
Public Types inherited from tvm::runtime::Object | |
typedef void(* | FDeleter) (Object *self) |
Object deleter. More... | |
using | RefCounterType = std::atomic< int32_t > |
Static Public Member Functions inherited from tvm::runtime::Object | |
static std::string | TypeIndex2Key (uint32_t tindex) |
Get the type key of the corresponding index from runtime. More... | |
static size_t | TypeIndex2KeyHash (uint32_t tindex) |
Get the type key hash of the corresponding index from runtime. More... | |
static uint32_t | TypeKey2Index (const std::string &key) |
Get the type index of the corresponding key from runtime. More... | |
static uint32_t | _GetOrAllocRuntimeTypeIndex () |
static uint32_t | RuntimeTypeIndex () |
Protected Member Functions inherited from tvm::runtime::Object | |
void | IncRef () |
developer function, increases reference counter. More... | |
void | DecRef () |
developer function, decrease reference counter. More... | |
Static Protected Member Functions inherited from tvm::runtime::Object | |
static uint32_t | GetOrAllocRuntimeTypeIndex (const std::string &key, uint32_t static_tindex, uint32_t parent_tindex, uint32_t type_child_slots, bool type_child_slots_can_overflow) |
Get the type index using type key. More... | |
Protected Attributes inherited from tvm::runtime::Object | |
uint32_t | type_index_ {0} |
Type index(tag) that indicates the type of the object. More... | |
RefCounterType | ref_counter_ {0} |
The internal reference counter. More... | |
FDeleter | deleter_ = nullptr |
deleter of this object to enable customized allocation. If the deleter is nullptr, no deletion will be performed. The creator of the object must always set the deleter field properly. More... | |
A frame that represents the block.
|
finalvirtual |
The method called when exiting RAII scope.
Reimplemented from tvm::script::ir_builder::IRBuilderFrameNode.
tvm::script::ir_builder::tir::BlockFrameNode::TVM_DECLARE_FINAL_OBJECT_INFO | ( | BlockFrameNode | , |
TIRFrameNode | |||
) |
|
inline |
|
staticconstexpr |
Array<tvm::tir::Buffer> tvm::script::ir_builder::tir::BlockFrameNode::alloc_buffers |
The buffer allocated in the block.
The annotation of the block.
Optional<tvm::tir::Stmt> tvm::script::ir_builder::tir::BlockFrameNode::init |
The init statement of the bolck.
The corresponding values of the iter vars.
Array<tvm::tir::IterVar> tvm::script::ir_builder::tir::BlockFrameNode::iter_vars |
The variables of the block.
Array<tvm::tir::MatchBufferRegion> tvm::script::ir_builder::tir::BlockFrameNode::match_buffers |
The match buffer regions.
String tvm::script::ir_builder::tir::BlockFrameNode::name |
The name of the block.
bool tvm::script::ir_builder::tir::BlockFrameNode::no_realize |
The flag whether to construct BlockRealize or Block.
The predicate of the block realization, the block will only be executed when the predicate is true.
Optional<Array<tvm::tir::BufferRegion> > tvm::script::ir_builder::tir::BlockFrameNode::reads |
The read buffer regions of the block.
Optional<Array<tvm::tir::BufferRegion> > tvm::script::ir_builder::tir::BlockFrameNode::writes |
The write buffer regions of the block.