tvm_tirx.tirx.script.alloc_scalar

Contents

tvm_tirx.tirx.script.alloc_scalar#

tvm_tirx.tirx.script.alloc_scalar(dtype='float32', scope=_SCOPE_GLOBAL, *, annotations=None)#

Allocate a zero-dimensional tensor (scalar), with optional allocation annotations.

Parameters:
  • dtype (str)

  • scope (str | MemoryScope)

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

Return type:

AlreadyEmitted[TensorLoad]