commit | 96e922e8ee4187906211ee0d1dd0f3e27667c170 | [log] [tgz] |
---|---|---|
author | Pablo Tello <pablo.tello@arm.com> | Wed Sep 26 11:25:15 2018 +0100 |
committer | Anthony Barbier <anthony.barbier@arm.com> | Fri Nov 02 16:55:19 2018 +0000 |
tree | 4e9458ef509687d45ce59fb135d170922ba5a783 | |
parent | 4bb17337e8330d01e469dfcb42220fe56b85a59f [diff] [blame] |
COMPMID-1602: Added Winograd tests for kernel 7x1 and 1x7. Change-Id: I9250b2e8020fe87c6ed4de582bbc7460bbd8e94b Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/150287 Tested-by: bsgcomp <bsgcomp@arm.com> Reviewed-by: Isabella Gottardi <isabella.gottardi@arm.com>
diff --git a/tests/validation/fixtures/WinogradConvolutionLayerFixture.h b/tests/validation/fixtures/WinogradConvolutionLayerFixture.h index 41f16d3..15ce201 100644 --- a/tests/validation/fixtures/WinogradConvolutionLayerFixture.h +++ b/tests/validation/fixtures/WinogradConvolutionLayerFixture.h
@@ -263,7 +263,17 @@ // Set output tile Size2D output_tile(4U, 4U); - if(weights_shape[0] == 1) + if(weights_shape[0] == 7 && weights_shape[1] == 1) + { + output_tile.width = 2; + output_tile.height = 1; + } + else if(weights_shape[0] == 1 && weights_shape[1] == 7) + { + output_tile.width = 1; + output_tile.height = 2; + } + else if(weights_shape[0] == 1) { output_tile.width = 1; }