Class TensorStoreObj

• Defined in File stmt.h

Inheritance Relationships

Base Type

• public tvm::StmtObj (Class StmtObj)

Class Documentation

class TensorStoreObj : public tvm::StmtObj

Store a primitive value to an indexed destination expression.

buffer[i, j] = value;

Public Functions

inline explicit TensorStoreObj(ffi::UnsafeInit tag)

inline TensorStoreObj(Expr dest, PrimExpr value)

Public Members

Expr dest

The indexed destination expression.

ffi::Array<PrimExpr> indices

The indices location to be stored.

PrimExpr value

The value to be stored.

Public Static Functions

static inline void RegisterReflection()

static inline int32_t _GetOrAllocRuntimeTypeIndex()

static inline int32_t RuntimeTypeIndex() noexcept