@@ -1745,7 +1745,7 @@ def setup_method(self):
17451745 self .random_stream = np .random .default_rng (utt .fetch_seed ())
17461746
17471747 self .inputs_shapes = [(8 , 1 , 12 , 12 ), (1 , 1 , 5 , 5 ), (1 , 1 , 5 , 6 ), (1 , 1 , 6 , 6 )]
1748- self .filters_shapes = [(5 , 1 , 2 , 2 ), (1 , 1 , 3 , 3 )]
1748+ self .filters_shapes = [(5 , 1 , 2 , 2 ), (1 , 1 , 3 , 3 )] * 2
17491749
17501750 self .subsamples = [(1 , 1 ), (2 , 2 )]
17511751 self .border_modes = ["valid" , "full" ]
@@ -1764,7 +1764,7 @@ def test_conv2d_grad_wrt_inputs(self):
17641764 # results are the same.
17651765
17661766 for in_shape , fltr_shape in zip (
1767- self .inputs_shapes , self .filters_shapes , strict = False
1767+ self .inputs_shapes , self .filters_shapes , strict = True
17681768 ):
17691769 for bm in self .border_modes :
17701770 for ss in self .subsamples :
@@ -1828,7 +1828,7 @@ def test_conv2d_grad_wrt_weights(self):
18281828 # results are the same.
18291829
18301830 for in_shape , fltr_shape in zip (
1831- self .inputs_shapes , self .filters_shapes , strict = False
1831+ self .inputs_shapes , self .filters_shapes , strict = True
18321832 ):
18331833 for bm in self .border_modes :
18341834 for ss in self .subsamples :
0 commit comments