tvm
Public Member Functions | Static Public Member Functions | Public Attributes | Static Public Attributes | List of all members
tvm::script::ir_builder::tir::AllocateFrameNode Class Reference

A frame represents the allocate. More...

#include <frame.h>

Inheritance diagram for tvm::script::ir_builder::tir::AllocateFrameNode:
Collaboration diagram for tvm::script::ir_builder::tir::AllocateFrameNode:

Public Member Functions

 TVM_DECLARE_FINAL_OBJECT_INFO (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_DECLARE_BASE_OBJECT_INFO (TIRFrameNode, IRBuilderFrameNode)
 
- Public Member Functions inherited from tvm::script::ir_builder::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 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

Array< PrimExprextents
 The extents of the allocate. More...
 
DataType dtype
 The data type of the buffer. More...
 
String storage_scope
 The storage scope. More...
 
PrimExpr condition
 The condition. More...
 
Map< 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
Array< tvm::tir::Stmtstmts
 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...
 

Static Public Attributes

static constexpr const char * _type_key = "script.ir_builder.tir.AllocateFrame"
 
- 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"
 

Detailed Description

A frame represents the allocate.

See also
AllocateFrame

Member Function Documentation

◆ ExitWithScope()

void tvm::script::ir_builder::tir::AllocateFrameNode::ExitWithScope ( )
finalvirtual

The method called when exiting RAII scope.

See also
tvm::support::With

Reimplemented from tvm::script::ir_builder::IRBuilderFrameNode.

◆ RegisterReflection()

static void tvm::script::ir_builder::tir::AllocateFrameNode::RegisterReflection ( )
inlinestatic

◆ TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::script::ir_builder::tir::AllocateFrameNode::TVM_DECLARE_FINAL_OBJECT_INFO ( AllocateFrameNode  ,
TIRFrameNode   
)

Member Data Documentation

◆ _type_key

constexpr const char* tvm::script::ir_builder::tir::AllocateFrameNode::_type_key = "script.ir_builder.tir.AllocateFrame"
staticconstexpr

◆ annotations

Map<String, Any> tvm::script::ir_builder::tir::AllocateFrameNode::annotations

Additional annotation hints.

◆ buffer_var

tvm::tir::Var tvm::script::ir_builder::tir::AllocateFrameNode::buffer_var

The buffer var.

◆ condition

PrimExpr tvm::script::ir_builder::tir::AllocateFrameNode::condition

The condition.

◆ dtype

DataType tvm::script::ir_builder::tir::AllocateFrameNode::dtype

The data type of the buffer.

◆ extents

Array<PrimExpr> tvm::script::ir_builder::tir::AllocateFrameNode::extents

The extents of the allocate.

◆ storage_scope

String tvm::script::ir_builder::tir::AllocateFrameNode::storage_scope

The storage scope.


The documentation for this class was generated from the following file: