@@ -2475,7 +2475,7 @@ def test_batch_to_spacend(self):
2475
2475
@check_opset_min_version (11 , "BatchToSpaceND" )
2476
2476
def test_batch_to_spacend_non_const_7d (self ):
2477
2477
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
2479
2479
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 ]]:
2480
2480
# test 1D upto 7D block shapes
2481
2481
for block_shape in [[2 , 3 ], [2 ]]:
@@ -2496,7 +2496,7 @@ def test_batch_to_spacend_non_const_7d(self):
2496
2496
@check_opset_min_version (11 , "SpaceToBatchND" )
2497
2497
def test_space_to_batchnd_non_const_7d (self ):
2498
2498
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
2500
2500
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 ]]:
2501
2501
# test 1D upto 7D block shapes
2502
2502
for block_shape in [[2 ], [2 , 2 ]]:
0 commit comments