From 4d2ca3a9cc65eb8253d69c54945e4bc97921439f Mon Sep 17 00:00:00 2001
From: Jay Choy <91728831+ZJay07@users.noreply.github.com>
Date: Wed, 27 Mar 2024 10:23:46 +0000
Subject: [PATCH] fixed tensorflow one_hot

---
 ivy/functional/frontends/tensorflow/general_functions.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/ivy/functional/frontends/tensorflow/general_functions.py b/ivy/functional/frontends/tensorflow/general_functions.py
index cc98ef66453e8..b2f85b7aa8a51 100644
--- a/ivy/functional/frontends/tensorflow/general_functions.py
+++ b/ivy/functional/frontends/tensorflow/general_functions.py
@@ -306,6 +306,9 @@ def norm(tensor, ord="euclidean", axis=None, keepdims=None, name=None):
     )
 
 
+@with_supported_dtypes(
+    {"2.15.0 and below": ("uint8", "int8", "int32", "int64")}, "tensorflow"
+)
 @to_ivy_arrays_and_back
 def one_hot(
     indices: ivy.Array,