#12730: bert slice support unit tests CORRECTIOn

tenstorrent · Sep 17, 2024 · f6fd4a1 · f6fd4a1
1 parent c860e7e
commit f6fd4a1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/ttnn/unit_tests/operations/test_slice.py b/tests/ttnn/unit_tests/operations/test_slice.py
@@ -412,8 +412,8 @@ def test_slice_negative_ends(layout, dim, ends, device):
     "input_shape, input_start, input_ends",
     (
         ((1, 1, 1, 256), (0, 0, 0, 0), (1, 1, 1, -1)),
-        ((1, 256), (0, 0), (-1, 1)),
-        ((1, 512), (0, 0), (-1, 1)),
+        ((1, 256), (0, 0), (-1, 256)),
+        ((1, 512), (0, 0), (-1, 512)),
         ((1, 512), (0, 0), (1, 256)),
     ),
 )