Skip to content

Commit 07dcd95

Browse files
authored
Arm Backend: Do not use None in Slice
Differential Revision: D83561950 Pull Request resolved: pytorch#14695
1 parent 70bffae commit 07dcd95

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

backends/arm/test/ops/test_slice.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,11 +34,11 @@
3434
test_data_suite = {
3535
"ones_neg_3": lambda: (torch.ones(10), [(3, -3)]),
3636
"ones_neg_8": lambda: (torch.ones(10), [(-8, 3)]),
37-
"ones_slice_2": lambda: (torch.ones(10, 10), [(1, 3), (3, None)]),
38-
"ones_slice_3": lambda: (torch.ones(10, 10, 10), [(0, 7), (0, None), (0, 8)]),
37+
"ones_slice_2": lambda: (torch.ones(10, 10), [(1, 3), (3, 10)]),
38+
"ones_slice_3": lambda: (torch.ones(10, 10, 10), [(0, 7), (0, 10), (0, 8)]),
3939
"ones_slice_4": lambda: (
4040
torch.ones((1, 12, 10, 10)),
41-
[(None, None), (None, 5), (3, 5), (4, 10)],
41+
[(0, 1), (0, 5), (3, 5), (4, 10)],
4242
),
4343
}
4444

0 commit comments

Comments
 (0)