tvm_tirx.tirx.script.alloc_local

Contents

tvm_tirx.tirx.script.alloc_local#

tvm_tirx.tirx.script.alloc_local(shape, dtype='float32', data=None, *, scope='local', align=-1, layout=<MISSING>, allocated_addr=None, annotations=None)#

Emit a tensor allocation and return its bound variable in a receipt.

Return type:

AlreadyEmitted[Var]