|
tvm
|
A frame that represents the block. More...
#include <frame.h>
Public Member Functions | |
| TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("script.ir_builder.tir.BlockFrame", 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_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::String | name |
| The name of the block. More... | |
| ffi::Array< tvm::tir::IterVar > | iter_vars |
| The variables of the block. More... | |
| ffi::Optional< ffi::Array< tvm::tir::BufferRegion > > | reads |
| The read buffer regions of the block. More... | |
| ffi::Optional< ffi::Array< tvm::tir::BufferRegion > > | writes |
| The write buffer regions of the block. More... | |
| ffi::Optional< tvm::tir::Stmt > | init |
| The init statement of the bolck. More... | |
| ffi::Array< tvm::tir::Buffer > | alloc_buffers |
| The buffer allocated in the block. More... | |
| ffi::Array< tvm::tir::MatchBufferRegion > | match_buffers |
| The match buffer regions. More... | |
| ffi::Optional< ffi::Map< ffi::String, Any > > | annotations |
| The annotation of the block. More... | |
| ffi::Array< PrimExpr > | iter_values |
| The corresponding values of the iter vars. More... | |
| ffi::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 | |
| 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 |
A frame that represents the block.
|
finalvirtual |
The method called when exiting RAII scope.
Reimplemented from tvm::script::ir_builder::IRBuilderFrameNode.
|
inlinestatic |
| tvm::script::ir_builder::tir::BlockFrameNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL | ( | "script.ir_builder.tir.BlockFrame" | , |
| BlockFrameNode | , | ||
| TIRFrameNode | |||
| ) |
| ffi::Array<tvm::tir::Buffer> tvm::script::ir_builder::tir::BlockFrameNode::alloc_buffers |
The buffer allocated in the block.
| ffi::Optional<ffi::Map<ffi::String, Any> > tvm::script::ir_builder::tir::BlockFrameNode::annotations |
The annotation of the block.
| ffi::Optional<tvm::tir::Stmt> tvm::script::ir_builder::tir::BlockFrameNode::init |
The init statement of the bolck.
| ffi::Array<PrimExpr> tvm::script::ir_builder::tir::BlockFrameNode::iter_values |
The corresponding values of the iter vars.
| ffi::Array<tvm::tir::IterVar> tvm::script::ir_builder::tir::BlockFrameNode::iter_vars |
The variables of the block.
| ffi::Array<tvm::tir::MatchBufferRegion> tvm::script::ir_builder::tir::BlockFrameNode::match_buffers |
The match buffer regions.
| ffi::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.
| ffi::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.
| ffi::Optional<ffi::Array<tvm::tir::BufferRegion> > tvm::script::ir_builder::tir::BlockFrameNode::reads |
The read buffer regions of the block.
| ffi::Optional<ffi::Array<tvm::tir::BufferRegion> > tvm::script::ir_builder::tir::BlockFrameNode::writes |
The write buffer regions of the block.