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

Load value from the high dimension buffer. More...

#include <expr.h>

Inheritance diagram for tvm::tir::BufferLoadNode:
Collaboration diagram for tvm::tir::BufferLoadNode:

Public Member Functions

 TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("tir.BufferLoad", BufferLoadNode, PrimExprNode)
 
- Public Member Functions inherited from tvm::PrimExprNode
 TVM_OBJECT_ENABLE_SCRIPT_PRINTER ()
 
 TVM_FFI_DECLARE_OBJECT_INFO ("ir.PrimExpr", PrimExprNode, BaseExprNode)
 
- Public Member Functions inherited from tvm::BaseExprNode
 TVM_FFI_DECLARE_OBJECT_INFO ("ir.BaseExpr", BaseExprNode, Object)
 

Static Public Member Functions

static void RegisterReflection ()
 
- Static Public Member Functions inherited from tvm::PrimExprNode
static void RegisterReflection ()
 
- Static Public Member Functions inherited from tvm::BaseExprNode
static void RegisterReflection ()
 

Public Attributes

Buffer buffer
 The buffer variable. More...
 
ffi::Array< PrimExprindices
 The indices location to be loaded. More...
 
ffi::Optional< PrimExprpredicate
 The predicate mask for loading values. More...
 
- Public Attributes inherited from tvm::PrimExprNode
DataType dtype
 The runtime data type of the primitive expression. More...
 
- Public Attributes inherited from tvm::BaseExprNode
Span span
 Span that points to the original source code. Reserved debug information. More...
 

Friends

class BufferLoad
 
class CustomDatatypesLowerer
 
class VectorTypeRewriter
 
class Vectorizer
 

Additional Inherited Members

- Static Public Attributes inherited from tvm::PrimExprNode
static constexpr const uint32_t _type_child_slots = 40
 
- Static Public Attributes inherited from tvm::BaseExprNode
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode
 
static constexpr const uint32_t _type_child_slots = 64
 

Detailed Description

Load value from the high dimension buffer.

value = buffer[i, j];
Buffer buffer
The buffer variable.
Definition: expr.h:536
See also
BufferStore

Member Function Documentation

◆ RegisterReflection()

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

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::tir::BufferLoadNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL ( "tir.BufferLoad"  ,
BufferLoadNode  ,
PrimExprNode   
)

Friends And Related Function Documentation

◆ BufferLoad

friend class BufferLoad
friend

◆ CustomDatatypesLowerer

friend class CustomDatatypesLowerer
friend

◆ Vectorizer

friend class Vectorizer
friend

◆ VectorTypeRewriter

friend class VectorTypeRewriter
friend

Member Data Documentation

◆ buffer

Buffer tvm::tir::BufferLoadNode::buffer

The buffer variable.

◆ indices

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

The indices location to be loaded.

◆ predicate

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

The predicate mask for loading values.


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