tvm_tirx.tirx.script.tensor_store

Contents

tvm_tirx.tirx.script.tensor_store#

tvm_tirx.tirx.script.tensor_store(dest, indices, value)#

Emit a tensor store and return a receipt for the stored statement.

Parameters:

  • dest (Expr) – The destination expression.

  • indices (List[Union[Expr, slice]]) – The indices location to be stored.

  • value (Expr) – The value to be stored.

Returns:

result (AlreadyEmitted[Stmt]) – Receipt for the exact stored statement; consuming it does not emit again.

Return type:

AlreadyEmitted[Stmt]

Contents