Store a primitive value to an indexed destination expression. More...

#include <stmt.h>

Inheritance diagram for tvm::TensorStoreNode:

Collaboration diagram for tvm::TensorStoreNode:

Public Member Functions
	TensorStoreNode (ffi::UnsafeInit tag)

	TensorStoreNode (Expr dest, PrimExpr value)

	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("ir.TensorStore", TensorStoreNode, StmtNode)

Public Member Functions inherited from tvm::StmtNode
	StmtNode ()=default

	StmtNode (Location loc)

	TVM_FFI_DECLARE_OBJECT_INFO ("ir.Stmt", StmtNode, ffi::Object)

Static Public Member Functions
static void	RegisterReflection ()

Static Public Member Functions inherited from tvm::StmtNode
static void	RegisterReflection ()

Public Attributes
Expr	dest
	The indexed destination expression.

ffi::Array< PrimExpr >	indices
	The indices location to be stored.

PrimExpr	value
	The value to be stored.

Public Attributes inherited from tvm::StmtNode
Location	loc = UnknownLoc()
	Location that points to the original source code. Reserved debug information.

Additional Inherited Members
Static Public Attributes inherited from tvm::StmtNode
static constexpr TVMFFISEqHashKind	_type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode

static constexpr bool	_type_s_eq_hash_subclass_kind_fixed = true

static constexpr const uint32_t	_type_child_slots = 15

Detailed Description

Store a primitive value to an indexed destination expression.

buffer[i, j] = value;

tvm::TensorStoreNode::value

PrimExpr value

The value to be stored.

Definition stmt.h:319

Constructor & Destructor Documentation

◆ TensorStoreNode() [1/2]

tvm::TensorStoreNode::TensorStoreNode ( ffi::UnsafeInit tag )

inlineexplicit

◆ TensorStoreNode() [2/2]

tvm::TensorStoreNode::TensorStoreNode	(	Expr	dest,
		PrimExpr	value
	)

inline

Member Function Documentation

◆ RegisterReflection()

static void tvm::TensorStoreNode::RegisterReflection ( )

inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::TensorStoreNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"ir.TensorStore"	,
		TensorStoreNode	,
		StmtNode
	)

Member Data Documentation

◆ dest

Expr tvm::TensorStoreNode::dest

The indexed destination expression.

◆ indices

ffi::Array<PrimExpr> tvm::TensorStoreNode::indices

The indices location to be stored.

◆ value

PrimExpr tvm::TensorStoreNode::value