tvm
Loading...
Searching...
No Matches
Public Member Functions | List of all members
tvm::TensorStore Class Reference

Managed reference to TensorStoreNode. More...

#include <stmt.h>

Inheritance diagram for tvm::TensorStore:
Collaboration diagram for tvm::TensorStore:

Public Member Functions

TVM_TIRX_DLL TensorStore (Expr dest, ffi::Array< PrimExpr > indices, PrimExpr value, Location loc=UnknownLoc())
 
 TensorStore (ffi::ObjectPtr< TensorStoreNode > node)
 
 TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (TensorStore, Stmt, TensorStoreNode)
 
 TVM_DEFINE_OBJECT_REF_COW_METHOD (TensorStoreNode)
 
- Public Member Functions inherited from tvm::Stmt
 Stmt (ffi::ObjectPtr< StmtNode > node)
 
 TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (Stmt, ffi::ObjectRef, StmtNode)
 

Detailed Description

Managed reference to TensorStoreNode.

See also
TensorStoreNode

Constructor & Destructor Documentation

◆ TensorStore() [1/2]

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

◆ TensorStore() [2/2]

tvm::TensorStore::TensorStore ( ffi::ObjectPtr< TensorStoreNode >  node)
inlineexplicit

Member Function Documentation

◆ TVM_DEFINE_OBJECT_REF_COW_METHOD()

tvm::TensorStore::TVM_DEFINE_OBJECT_REF_COW_METHOD ( TensorStoreNode  )

◆ TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE()

tvm::TensorStore::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE ( TensorStore  ,
Stmt  ,
TensorStoreNode   
)

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