Class TensorStore

Contents

Class TensorStore#

Defined in File stmt.h

Inheritance Relationships#

Base Type#

public tvm::Stmt (Class Stmt)

Class Documentation#

class TensorStore : public tvm::Stmt #

Managed reference to TensorStoreObj.

See also

Public Types

using ContainerType = TensorStoreObj #

Public Functions

explicit TensorStore(Expr dest, ffi::Array<PrimExpr> indices, PrimExpr value, Location loc = UnknownLoc())#

inline explicit TensorStore(ffi::ObjectPtr<TensorStoreObj> node)#

inline explicit TensorStore(::tvm::ffi::UnsafeInit tag)#

inline __PtrType operator->() const#

inline __PtrType get() const#

inline TensorStoreObj *CopyOnWrite()#