Function tvm::script::ir_builder::tirx::TensorStore

Function tvm::script::ir_builder::tirx::TensorStore#

Function Documentation#

tvm::Stmt tvm::script::ir_builder::tirx::TensorStore(Expr dest, ffi::Array<PrimExpr> indices, PrimExpr value)#

Store data in a tensor destination.

Parameters:

  • dest – The destination expression.

  • indices – The indices location to be stored.

  • value – The value to be stored.

Returns:

The same statement that was added to the parent frame.