Skip to content

Commit 5d2c7a2

Browse files
committed
Minor fix
1 parent 672ed43 commit 5d2c7a2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_backend.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2475,7 +2475,7 @@ def test_batch_to_spacend(self):
24752475
@check_opset_min_version(11, "BatchToSpaceND")
24762476
def test_batch_to_spacend_non_const_7d(self):
24772477
x_type, y_type, z_type = np.int64, np.int64, np.int64
2478-
# test 3D upto 6D input tensors
2478+
# test 3D upto 7D input tensors
24792479
for x_shape in [[12, 4, 4], [12, 4, 8, 3], [12, 4, 8, 3, 2], [12, 4, 8, 3, 2, 3], [12, 4, 8, 3, 2, 1, 3]]:
24802480
# test 1D upto 7D block shapes
24812481
for block_shape in [[2, 3], [2]]:
@@ -2496,7 +2496,7 @@ def test_batch_to_spacend_non_const_7d(self):
24962496
@check_opset_min_version(11, "SpaceToBatchND")
24972497
def test_space_to_batchnd_non_const_7d(self):
24982498
x_type, y_type, z_type = np.int64, np.int64, np.int64
2499-
# test 3D upto 6D input tensors
2499+
# test 3D upto 7D input tensors
25002500
for x_shape in [[2, 4, 4], [1, 4, 8, 3], [1, 4, 8, 3, 2], [1, 4, 8, 3, 2, 3], [1, 4, 8, 3, 2, 1, 3]]:
25012501
# test 1D upto 7D block shapes
25022502
for block_shape in [[2], [2, 2]]:

0 commit comments

Comments
 (0)