aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/fixtures
diff options
context:
space:
mode:
authorPablo Tello <pablo.tello@arm.com>2018-09-26 11:25:15 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:55:19 +0000
commit96e922e8ee4187906211ee0d1dd0f3e27667c170 (patch)
tree4e9458ef509687d45ce59fb135d170922ba5a783 /tests/validation/fixtures
parent4bb17337e8330d01e469dfcb42220fe56b85a59f (diff)
downloadComputeLibrary-96e922e8ee4187906211ee0d1dd0f3e27667c170.tar.gz
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>
Diffstat (limited to 'tests/validation/fixtures')
-rw-r--r--tests/validation/fixtures/WinogradConvolutionLayerFixture.h12
1 files changed, 11 insertions, 1 deletions
diff --git a/tests/validation/fixtures/WinogradConvolutionLayerFixture.h b/tests/validation/fixtures/WinogradConvolutionLayerFixture.h
index 41f16d3a39..15ce201222 100644
--- a/tests/validation/fixtures/WinogradConvolutionLayerFixture.h
+++ b/tests/validation/fixtures/WinogradConvolutionLayerFixture.h
@@ -263,7 +263,17 @@ protected:
// 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;
}