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)) +)