tvm_tirx.tirx.script.assign

Contents

tvm_tirx.tirx.script.assign#

tvm_tirx.tirx.script.assign(dst, value)#

Store a value into scalar storage and return its emission receipt.

Parameters:
  • dst (TensorLoad | Var) – A tensor load identifying the destination tensor and indices, or a tensor variable with constant shape (1,). Immutable scalar variables and whole multi-element tensors are not assignment destinations.

  • value (Any) – The scalar value to store, converted by the ordinary tensor-store API.

Returns:

result (AlreadyEmitted[Stmt]) – Receipt for the exact emitted store. Parser consumption does not emit it again. Active builder source locations are preserved.

Return type:

AlreadyEmitted[Stmt]