Exposed set functions to dpctl.tensor namespace

Author: oleksandr-pavlyk

dpctl/tensor/__init__.py

@@ -179,6 +179,12 @@
     reduce_hypot,
     sum,
 )
+from ._set_functions import (
+    unique_all,
+    unique_counts,
+    unique_inverse,
+    unique_values,
+)
 from ._sorting import argsort, sort
 from ._testing import allclose
 
@@ -349,4 +355,8 @@
     "angle",
     "sort",
     "argsort",
+    "unique_all",
+    "unique_counts",
+    "unique_inverse",
+    "unique_values",
 ]

dpctl/tensor/_set_functions.py

@@ -15,6 +15,16 @@
 )
 from ._tensor_sorting_impl import _argsort_ascending, _sort_ascending
 
+__all__ = [
+    "unique_values",
+    "unique_counts",
+    "unique_inverse",
+    "unique_all",
+    "UniqueAllResult",
+    "UniqueCountsResult",
+    "UniqueInverseResult",
+]
+
 
 class UniqueAllResult(NamedTuple):
     values: dpt.usm_ndarray