fixed tensorflow.keras.activations.relu

ivy-llc · Mar 18, 2024 · b86cafa · b86cafa
1 parent dc14b57
commit b86cafa
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/ivy/functional/frontends/tensorflow/keras/activations.py b/ivy/functional/frontends/tensorflow/keras/activations.py
@@ -1,7 +1,7 @@
 import ivy
 import ivy.functional.frontends.tensorflow as tf_frontend
 from ivy.functional.frontends.tensorflow.func_wrapper import to_ivy_arrays_and_back
-from ivy import with_supported_dtypes
+from ivy import with_supported_dtypes, with_unsupported_dtypes
 
 
 ACTIVATION_FUNCTIONS = [
@@ -112,6 +112,10 @@ def linear(x):
     return ivy.array(x)
 
 
+@with_unsupported_dtypes(
+    {"2.15.0 and below": ("complex",)},
+    "tensorflow",
+)
 @to_ivy_arrays_and_back
 def relu(x, alpha=0.0, max_value=None, threshold=0.0):
     return ivy.relu(x)