We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
DLDeviceType
dpctl.tensor
1 parent fe71e99 commit 85d417fCopy full SHA for 85d417f
1 file changed
dpctl/tensor/__init__.py
@@ -93,7 +93,7 @@
93
from dpctl.tensor._reshape import reshape
94
from dpctl.tensor._search_functions import where
95
from dpctl.tensor._statistical_functions import mean, std, var
96
-from dpctl.tensor._usmarray import usm_ndarray
+from dpctl.tensor._usmarray import DLDeviceType, usm_ndarray
97
from dpctl.tensor._utility_functions import all, any, diff
98
99
from ._accumulation import cumulative_logsumexp, cumulative_prod, cumulative_sum
@@ -376,4 +376,5 @@
376
"nextafter",
377
"diff",
378
"count_nonzero",
379
+ "DLDeviceType",
380
]
0 commit comments