tvm_tirx.tirx.script.shared_scalar

Contents

tvm_tirx.tirx.script.shared_scalar#

tvm_tirx.tirx.script.shared_scalar(dtype='float32', *, annotations=None)#

Allocate a zero-dimensional tensor in shared memory.

Parameters:
  • dtype (str)

  • annotations (dict[str, Any] | None)

Return type:

AlreadyEmitted[TensorLoad]