From dc55d34c56dc34c4511a9ab769292e23caf1653a Mon Sep 17 00:00:00 2001
From: Janindu Pathirana <85122582+janchilling@users.noreply.github.com>
Date: Mon, 31 Jul 2023 11:29:30 +0530
Subject: [PATCH] #19402. Added Einsum function to rawops (#19405)

---
 .../frontends/tensorflow/raw_ops.py           | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/ivy/functional/frontends/tensorflow/raw_ops.py b/ivy/functional/frontends/tensorflow/raw_ops.py
index 953922b33123e..3b3d1eef02c78 100644
--- 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))
+)