updated handle_device_shifting for newly added functions

ivy-llc · Sep 25, 2023 · 48cc8d5 · 48cc8d5
1 parent 248f084
commit 48cc8d5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ivy/functional/ivy/experimental/layers.py b/ivy/functional/ivy/experimental/layers.py
@@ -3031,7 +3031,7 @@ def _cal_output_shape(
 @handle_partial_mixed_function
 @to_native_arrays_and_back
 @inputs_to_ivy_arrays
-@handle_device_shifting
+@handle_device
 def max_unpool1d(
     input: ivy.Array,
     indices: ivy.Array,
@@ -3110,7 +3110,7 @@ def max_unpool1d(
     "to_add": (
         "handle_backend_invalid",
         "to_native_arrays_and_back",
-        "handle_device_shifting",
+        "handle_device",
     ),
     "to_skip": ("inputs_to_ivy_arrays", "handle_partial_mixed_function"),
 }