tvm
|
A frame represents the allocate constant. More...
#include <frame.h>
Public Member Functions | |
TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("script.ir_builder.tir.AllocateConstFrame", AllocateConstFrameNode, TIRFrameNode) | |
void | ExitWithScope () final |
The method called when exiting RAII scope. More... | |
![]() | |
TVM_FFI_DECLARE_OBJECT_INFO ("script.ir_builder.tir.TIRFrame", TIRFrameNode, 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 void | RegisterReflection () |
![]() | |
static void | RegisterReflection () |
Public Attributes | |
DataType | dtype |
The data type of the buffer. More... | |
ffi::Array< PrimExpr > | extents |
The extents of the allocate. More... | |
tvm::runtime::Tensor | data |
The data associated with the constant. More... | |
tvm::tir::Var | buffer_var |
The buffer var. More... | |
ffi::Map< ffi::String, Any > | annotations |
Additional annotations about the allocation. More... | |
![]() | |
ffi::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... | |
Additional Inherited Members | |
![]() | |
static constexpr const bool | _type_mutable = true |
A frame represents the allocate constant.
|
finalvirtual |
The method called when exiting RAII scope.
Reimplemented from tvm::script::ir_builder::IRBuilderFrameNode.
|
inlinestatic |
tvm::script::ir_builder::tir::AllocateConstFrameNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL | ( | "script.ir_builder.tir.AllocateConstFrame" | , |
AllocateConstFrameNode | , | ||
TIRFrameNode | |||
) |
ffi::Map<ffi::String, Any> tvm::script::ir_builder::tir::AllocateConstFrameNode::annotations |
Additional annotations about the allocation.
tvm::tir::Var tvm::script::ir_builder::tir::AllocateConstFrameNode::buffer_var |
The buffer var.
tvm::runtime::Tensor tvm::script::ir_builder::tir::AllocateConstFrameNode::data |
The data associated with the constant.
DataType tvm::script::ir_builder::tir::AllocateConstFrameNode::dtype |
The data type of the buffer.
ffi::Array<PrimExpr> tvm::script::ir_builder::tir::AllocateConstFrameNode::extents |
The extents of the allocate.