|
tvm
|
A frame represents the allocate. More...
#include <frame.h>
Public Member Functions | |
| TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("script.ir_builder.tir.AllocateFrame", AllocateFrameNode, 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::Array< PrimExpr > | extents |
| The extents of the allocate. More... | |
| DataType | dtype |
| The data type of the buffer. More... | |
| ffi::String | storage_scope |
| The storage scope. More... | |
| PrimExpr | condition |
| The condition. More... | |
| ffi::Map< ffi::String, Any > | annotations |
| Additional annotation hints. More... | |
| tvm::tir::Var | buffer_var |
| The buffer var. 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 represents the allocate.
|
finalvirtual |
The method called when exiting RAII scope.
Reimplemented from tvm::script::ir_builder::IRBuilderFrameNode.
|
inlinestatic |
| tvm::script::ir_builder::tir::AllocateFrameNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL | ( | "script.ir_builder.tir.AllocateFrame" | , |
| AllocateFrameNode | , | ||
| TIRFrameNode | |||
| ) |
| ffi::Map<ffi::String, Any> tvm::script::ir_builder::tir::AllocateFrameNode::annotations |
Additional annotation hints.
| tvm::tir::Var tvm::script::ir_builder::tir::AllocateFrameNode::buffer_var |
The buffer var.
| PrimExpr tvm::script::ir_builder::tir::AllocateFrameNode::condition |
The condition.
| DataType tvm::script::ir_builder::tir::AllocateFrameNode::dtype |
The data type of the buffer.
| ffi::Array<PrimExpr> tvm::script::ir_builder::tir::AllocateFrameNode::extents |
The extents of the allocate.
| ffi::String tvm::script::ir_builder::tir::AllocateFrameNode::storage_scope |
The storage scope.