@@ -1745,7 +1745,7 @@ def setup_method(self):
1745
1745
self .random_stream = np .random .default_rng (utt .fetch_seed ())
1746
1746
1747
1747
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
1749
1749
1750
1750
self .subsamples = [(1 , 1 ), (2 , 2 )]
1751
1751
self .border_modes = ["valid" , "full" ]
@@ -1764,7 +1764,7 @@ def test_conv2d_grad_wrt_inputs(self):
1764
1764
# results are the same.
1765
1765
1766
1766
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
1768
1768
):
1769
1769
for bm in self .border_modes :
1770
1770
for ss in self .subsamples :
@@ -1828,7 +1828,7 @@ def test_conv2d_grad_wrt_weights(self):
1828
1828
# results are the same.
1829
1829
1830
1830
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
1832
1832
):
1833
1833
for bm in self .border_modes :
1834
1834
for ss in self .subsamples :
0 commit comments