tvm
|
#include <frame.h>
Public Member Functions | |
TVM_DECLARE_FINAL_OBJECT_INFO (DeclBufferFrameNode, TIRFrameNode) | |
void | ExitWithScope () final |
The method called when exiting RAII scope. More... | |
![]() | |
TVM_DECLARE_BASE_OBJECT_INFO (TIRFrameNode, 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 void | RegisterReflection () |
![]() | |
static void | RegisterReflection () |
Public Attributes | |
tvm::tir::Buffer | buffer |
The declared buffer. More... | |
bool | allocated |
The buffer allocated or not. More... | |
![]() | |
Array< tvm::tir::Stmt > | stmts |
The Stmt within in this frame. More... | |
![]() | |
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.DeclBufferFrame" |
![]() | |
static constexpr const char * | _type_key = "script.ir_builder.tir.TIRFrame" |
![]() | |
static constexpr const char * | _type_key = "script.ir_builder.IRBuilderFrame" |
|
finalvirtual |
The method called when exiting RAII scope.
Reimplemented from tvm::script::ir_builder::IRBuilderFrameNode.
|
inlinestatic |
tvm::script::ir_builder::tir::DeclBufferFrameNode::TVM_DECLARE_FINAL_OBJECT_INFO | ( | DeclBufferFrameNode | , |
TIRFrameNode | |||
) |
|
staticconstexpr |
bool tvm::script::ir_builder::tir::DeclBufferFrameNode::allocated |
The buffer allocated or not.
tvm::tir::Buffer tvm::script::ir_builder::tir::DeclBufferFrameNode::buffer |
The declared buffer.