tvm_tirx.runtime.empty

Contents

tvm_tirx.runtime.empty#

tvm_tirx.runtime.empty(shape, dtype='float32', device=None, mem_scope=None)#

Create an empty array given shape and device

Parameters:

  • shape (Union[tvm_ffi.Shape, Sequence[SupportsInt]]) – The shape of the array.

  • dtype (type or str) – The data type of the array.

  • device (Device) – The device of the array.

  • mem_scope (Optional[str or tvm_tirx.runtime.MemoryScope]) – The memory scope of the array.

Returns:

arr (tvm_tirx.runtime.Tensor) – The array tvm supported.

Contents