#19402. Added Einsum function to rawops (#19405)

ivy-llc · Jul 31, 2023 · dc55d34 · dc55d34
1 parent 216a696
commit dc55d34
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/ivy/functional/frontends/tensorflow/raw_ops.py b/ivy/functional/frontends/tensorflow/raw_ops.py
@@ -938,3 +938,22 @@ def Svd(*, input, full_matrices=False, compute_uv=True, name=None):
         "complex128",
     ),
 }
+
+
+Einsum = to_ivy_arrays_and_back(
+    with_supported_dtypes(
+        {
+            "2.13.0 and below": (
+                "bfloat16",
+                "complex128 ",
+                "complex64",
+                "float64",
+                "float32",
+                "float16",
+                "int64",
+                "int32"
+            ),
+        },
+        "tensorflow",
+    )(map_raw_ops_alias(tf_frontend.general_functions.einsum))
+)