diff --git a/mace/ops/winograd_convolution_test.cc b/mace/ops/winograd_convolution_test.cc index ae87a81817ca282f8fc773e0223ff2245a09e342..3e594500d86bf984452c37f36dee7bd59580f8ef 100644 --- a/mace/ops/winograd_convolution_test.cc +++ b/mace/ops/winograd_convolution_test.cc @@ -281,63 +281,63 @@ void WinogradConvolutionWithPad(const index_t batch, TEST_F(WinogradConvlutionTest, AlignedConvolutionM2WithPad) { WinogradConvolutionWithPad(1, 32, 32, 32, 16, 1, 2); - WinogradConvolutionWithPad(1, 32, 32, 32, 16, + WinogradConvolutionWithPad(1, 32, 32, 32, 16, 2, 2); } TEST_F(WinogradConvlutionTest, UnAlignedConvolutionM2WithPad) { WinogradConvolutionWithPad(1, 61, 67, 31, 37, 1, 2); - WinogradConvolutionWithPad(1, 61, 67, 37, 31, + WinogradConvolutionWithPad(1, 61, 67, 37, 31, 2, 2); } TEST_F(WinogradConvlutionTest, BatchConvolutionWithM2Pad) { WinogradConvolutionWithPad(3, 64, 64, 32, 32, 1, 2); - WinogradConvolutionWithPad(5, 61, 67, 37, 31, + WinogradConvolutionWithPad(5, 61, 67, 37, 31, 2, 2); } TEST_F(WinogradConvlutionTest, AlignedConvolutionM6WithPad) { WinogradConvolutionWithPad(1, 32, 32, 32, 16, 1, 6); - WinogradConvolutionWithPad(1, 32, 32, 32, 16, + WinogradConvolutionWithPad(1, 32, 32, 32, 16, 2, 6); } TEST_F(WinogradConvlutionTest, UnAlignedConvolutionM6WithPad) { WinogradConvolutionWithPad(1, 61, 67, 31, 37, 1, 6); - WinogradConvolutionWithPad(1, 61, 67, 37, 31, + WinogradConvolutionWithPad(1, 61, 67, 37, 31, 2, 6); } TEST_F(WinogradConvlutionTest, BatchConvolutionWithM6Pad) { WinogradConvolutionWithPad(3, 64, 64, 32, 32, 1, 6); -// WinogradConvolutionWithPad(5, 61, 67, 37, 31, -// 2, 6); + WinogradConvolutionWithPad(5, 61, 67, 37, 31, + 2, 6); } TEST_F(WinogradConvlutionTest, AlignedConvolutionM4WithPad) { WinogradConvolutionWithPad(1, 32, 32, 32, 16, 1, 4); - WinogradConvolutionWithPad(1, 32, 32, 32, 16, + WinogradConvolutionWithPad(1, 32, 32, 32, 16, 2, 4); } TEST_F(WinogradConvlutionTest, UnAlignedConvolutionM4WithPad) { WinogradConvolutionWithPad(1, 61, 67, 31, 37, 1, 4); - WinogradConvolutionWithPad(1, 61, 67, 37, 31, + WinogradConvolutionWithPad(1, 61, 67, 37, 31, 2, 4); } TEST_F(WinogradConvlutionTest, BatchConvolutionWithM4Pad) { WinogradConvolutionWithPad(3, 64, 64, 32, 32, 1, 4); - WinogradConvolutionWithPad(5, 61, 67, 37, 31, + WinogradConvolutionWithPad(5, 61, 67, 37, 31, 2, 4); }