numba_dppy.dpctl_iface.usm_ndarray_type module

class numba_dppy.dpctl_iface.usm_ndarray_type.USMNdArrayType(*args, **kwargs)

Bases: numba_dppy.dppy_array_type.DPPYArray

USMNdArrayType(dtype, ndim, layout, usm_type,

readonly=False, name=None, aligned=True, addrspace=None)

creates Numba type to represent dpctl.tensor.usm_ndarray.

copy(*args, **kwargs)

numba_dppy.dpctl_iface.usm_ndarray_type.typeof_usm_ndarray(val, c)

This function creates the Numba type (USMNdArrayType) when a usm_ndarray is passed.