fix: specify the int supported type for one_hot in tf frontend (#28699)

ivy-llc · Apr 2, 2024 · 619cce9 · 619cce9
1 parent 6a34ee1
commit 619cce9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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,