tvm_tirx.tirx.script.alloc_shared

Contents

tvm_tirx.tirx.script.alloc_shared#

tvm_tirx.tirx.script.alloc_shared(shape, dtype='float32', data=None, *, scope='shared', 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]