tvm
Public Member Functions | Static Public Member Functions | Public Attributes | List of all members
tvm::tir::BufferStoreNode Class Reference

Store value to the high dimension buffer. More...

#include <stmt.h>

Inheritance diagram for tvm::tir::BufferStoreNode:
Collaboration diagram for tvm::tir::BufferStoreNode:

Public Member Functions

 TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("tir.BufferStore", BufferStoreNode, StmtNode)
 
- Public Member Functions inherited from tvm::tir::StmtNode
 StmtNode ()=default
 
 StmtNode (Span span)
 
 TVM_OBJECT_ENABLE_SCRIPT_PRINTER ()
 
 TVM_FFI_DECLARE_OBJECT_INFO ("tir.Stmt", StmtNode, Object)
 

Static Public Member Functions

static void RegisterReflection ()
 
- Static Public Member Functions inherited from tvm::tir::StmtNode
static void RegisterReflection ()
 

Public Attributes

Buffer buffer
 The buffer variable. More...
 
PrimExpr value
 The value to be stored. More...
 
ffi::Array< PrimExprindices
 The indices location to be stored. More...
 
ffi::Optional< PrimExprpredicate
 The predicate mask for storing values. More...
 
- Public Attributes inherited from tvm::tir::StmtNode
Span span
 Span that points to the original source code. Reserved debug information. More...
 

Additional Inherited Members

- Static Public Attributes inherited from tvm::tir::StmtNode
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode
 
static constexpr const uint32_t _type_child_slots = 15
 

Detailed Description

Store value to the high dimension buffer.

buffer[i, j] = value;
Buffer buffer
The buffer variable.
Definition: stmt.h:197
PrimExpr value
The value to be stored.
Definition: stmt.h:199
See also
BufferLoad

Member Function Documentation

◆ RegisterReflection()

static void tvm::tir::BufferStoreNode::RegisterReflection ( )
inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::tir::BufferStoreNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL ( "tir.BufferStore"  ,
BufferStoreNode  ,
StmtNode   
)

Member Data Documentation

◆ buffer

Buffer tvm::tir::BufferStoreNode::buffer

The buffer variable.

◆ indices

ffi::Array<PrimExpr> tvm::tir::BufferStoreNode::indices

The indices location to be stored.

◆ predicate

ffi::Optional<PrimExpr> tvm::tir::BufferStoreNode::predicate

The predicate mask for storing values.

◆ value

PrimExpr tvm::tir::BufferStoreNode::value

The value to be stored.


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