Move ti.atan()/atanh() and reuse them

Author: vlad-perevezentsev

dpctl_ext/tensor/__init__.py

@@ -90,6 +90,8 @@
     angle,
     asin,
     asinh,
+    atan,
+    atanh,
 )
 from ._reduction import (
     argmax,
@@ -129,6 +131,8 @@
     "asinh",
     "asnumpy",
     "astype",
+    "atan",
+    "atanh",
     "broadcast_arrays",
     "broadcast_to",
     "can_cast",

dpctl_ext/tensor/_elementwise_funcs.py

@@ -178,6 +178,64 @@
 )
 del _asinh_docstring
 
+# U06: ===== ATAN  (x)
+_atan_docstring = r"""
+atan(x, /, \*, out=None, order='K')
+
+Computes inverse tangent for each element `x_i` for input array `x`.
+
+Args:
+    x (usm_ndarray):
+        Input array, expected to have a floating-point data type.
+    out (Union[usm_ndarray, None], optional):
+        Output array to populate.
+        Array must have the correct shape and the expected data type.
+    order ("C","F","A","K", optional):
+        Memory layout of the new output array, if parameter
+        `out` is ``None``.
+        Default: "K".
+
+Returns:
+    usm_ndarray:
+        An array containing the element-wise inverse tangent, in radians
+        and in the closed interval :math:`[-\pi/2, \pi/2]`. The data type
+        of the returned array is determined by the Type Promotion Rules.
+"""
+
+atan = UnaryElementwiseFunc(
+    "atan", ti._atan_result_type, ti._atan, _atan_docstring
+)
+del _atan_docstring
+
+# U07: ===== ATANH (x)
+_atanh_docstring = r"""
+atanh(x, /, \*, out=None, order='K')
+
+Computes hyperbolic inverse tangent for each element `x_i` for input array `x`.
+
+Args:
+    x (usm_ndarray):
+        Input array, expected to have a floating-point data type.
+    out (Union[usm_ndarray, None], optional):
+        Output array to populate.
+        Array must have the correct shape and the expected data type.
+    order ("C","F","A","K", optional):
+        Memory layout of the new output array, if parameter
+        `out` is ``None``.
+        Default: "K".
+
+Returns:
+    usm_ndarray:
+        An array containing the element-wise hyperbolic inverse tangent, in
+        radians. The data type of the returned array is determined by
+        the Type Promotion Rules.
+"""
+
+atanh = UnaryElementwiseFunc(
+    "atanh", ti._atanh_result_type, ti._atanh, _atanh_docstring
+)
+del _atanh_docstring
+
 # U43: ==== ANGLE        (x)
 _angle_docstring = r"""
 angle(x, /, \*, out=None, order='K')

dpnp/dpnp_iface_trigonometric.py

@@ -481,8 +481,8 @@ def _get_accumulation_res_dt(a, dtype):
 
 atan = DPNPUnaryFunc(
     "atan",
-    ti._atan_result_type,
-    ti._atan,
+    ti_ext._atan_result_type,
+    ti_ext._atan,
     _ATAN_DOCSTRING,
     mkl_fn_to_call="_mkl_atan_to_call",
     mkl_impl_fn="_atan",
@@ -656,8 +656,8 @@ def _get_accumulation_res_dt(a, dtype):
 
 atanh = DPNPUnaryFunc(
     "atanh",
-    ti._atanh_result_type,
-    ti._atanh,
+    ti_ext._atanh_result_type,
+    ti_ext._atanh,
     _ATANH_DOCSTRING,
     mkl_fn_to_call="_mkl_atanh_to_call",
     mkl_impl_fn="_atanh",