tvm
|
Node to represent a buffer. More...
#include <buffer.h>
Public Member Functions | |
BufferNode () | |
constructor More... | |
DataType | DefaultIndexType () const |
Array< PrimExpr > | ElemOffset (Array< PrimExpr > index) const |
Determine the offset in the buffer of the given index. More... | |
TVM_DECLARE_FINAL_OBJECT_INFO (BufferNode, Object) | |
TVM_OBJECT_ENABLE_SCRIPT_PRINTER () | |
Static Public Member Functions | |
static void | RegisterReflection () |
Public Attributes | |
Var | data |
The pointer to the head of the data. More... | |
DataType | dtype |
data type in the content of the tensor More... | |
Array< PrimExpr > | shape |
The type of the buffer prior to flattening. More... | |
Array< IntImm > | axis_separators |
Separators between input axes when generating flattened output axes. More... | |
Array< PrimExpr > | strides |
The strides of each dimension This can be an empty array, indicating array is contiguous. More... | |
PrimExpr | elem_offset |
The offset in terms of number of dtype elements (including lanes) More... | |
String | name |
optional name of the buffer More... | |
int | data_alignment |
Alignment requirement of data pointer in bytes. More... | |
int | offset_factor |
Factor of elem_offset field, elem_offset is guaranteed to be multiple of offset_factor. More... | |
BufferType | buffer_type |
buffer type More... | |
Span | span |
Span that points to the original source code. Reserved debug information. More... | |
Static Public Attributes | |
static constexpr const char * | _type_key = "tir.Buffer" |
static constexpr TVMFFISEqHashKind | _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode |
Node to represent a buffer.
|
inline |
constructor
|
inline |
Determine the offset in the buffer of the given index.
Returns the buffer offset, in number of elements of type dtype, without adjusting for number of lanes. (e.g. The number of float16x4 elements in a buffer of type float16x4.)
|
inlinestatic |
tvm::tir::BufferNode::TVM_DECLARE_FINAL_OBJECT_INFO | ( | BufferNode | , |
Object | |||
) |
tvm::tir::BufferNode::TVM_OBJECT_ENABLE_SCRIPT_PRINTER | ( | ) |
|
staticconstexpr |
|
staticconstexpr |
Array<IntImm> tvm::tir::BufferNode::axis_separators |
Separators between input axes when generating flattened output axes.
For buffers representing flat 1-d memory (e.g. any buffer in RAM), this should be an empty array. For buffers representing non-flat memory, each entry in axis_separators should be the first input axis that is part of a new flattened axis.
BufferType tvm::tir::BufferNode::buffer_type |
buffer type
Var tvm::tir::BufferNode::data |
The pointer to the head of the data.
int tvm::tir::BufferNode::data_alignment |
Alignment requirement of data pointer in bytes.
DataType tvm::tir::BufferNode::dtype |
data type in the content of the tensor
PrimExpr tvm::tir::BufferNode::elem_offset |
The offset in terms of number of dtype elements (including lanes)
String tvm::tir::BufferNode::name |
optional name of the buffer
int tvm::tir::BufferNode::offset_factor |
Factor of elem_offset field, elem_offset is guaranteed to be multiple of offset_factor.
Array<PrimExpr> tvm::tir::BufferNode::shape |
The type of the buffer prior to flattening.
This contains the shape as it is accessed by BufferLoad/BufferStore nodes, and used by the low-level code generators.
|
mutable |
Span that points to the original source code. Reserved debug information.
Array<PrimExpr> tvm::tir::BufferNode::strides |
The strides of each dimension This can be an empty array, indicating array is contiguous.