tvm_tirx.tirx.address_of

Contents

tvm_tirx.tirx.address_of#

tvm_tirx.tirx.address_of(obj, loc=UNKNOWN_LOC, *, ty=None)#

Returns the address of a tensor element or addressable variable.

Parameters:
  • obj (Union[Var, TensorLoad]) – The tensor, tensor load, or addressable variable.

  • loc (Location, optional) – The location of this operator in the source code.

Returns:

call (Expr) – The call expression.

Return type:

Expr