fix: specify supported types for softplus in tensorflow frontend (#28466

)
ivy-llc · Mar 26, 2024 · 9d65596 · 9d65596
1 parent 56a06fd
commit 9d65596
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/ivy/functional/backends/paddle/activations.py b/ivy/functional/backends/paddle/activations.py
@@ -117,8 +117,18 @@ def softmax(
     return paddle.divide(exp_x, paddle.sum(exp_x, axis=axis, keepdim=True))
 
 
-@with_unsupported_device_and_dtypes(
-    {"2.6.0 and below": {"cpu": ("float16", "bfloat16")}}, backend_version
+@with_supported_dtypes(
+    {
+        "2.6.0 and below": (
+            "int32",
+            "int64",
+            "float64",
+            "complex128",
+            "float32",
+            "complex64",
+        )
+    },
+    backend_version,
 )
 def softplus(
     x: paddle.Tensor,