From d267b05aaaec9b462a8c988c7b5fcebd5776c72f Mon Sep 17 00:00:00 2001 From: Pablo Tello Date: Mon, 19 Feb 2018 16:46:03 +0000 Subject: COMPMID-784: Fixed SAME paddding in WinogradLayer There were mismatches when using kernel size 5 and padding = SAME Change-Id: Id834e96ebcf665616f99c995b48e302dcff8dc48 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/121144 Tested-by: Jenkins Reviewed-by: Anthony Barbier --- tests/datasets/SmallConvolutionLayerDataset.h | 1 + 1 file changed, 1 insertion(+) (limited to 'tests/datasets/SmallConvolutionLayerDataset.h') diff --git a/tests/datasets/SmallConvolutionLayerDataset.h b/tests/datasets/SmallConvolutionLayerDataset.h index d88efbd21b..adb61de8e2 100644 --- a/tests/datasets/SmallConvolutionLayerDataset.h +++ b/tests/datasets/SmallConvolutionLayerDataset.h @@ -51,6 +51,7 @@ public: // Kernel size 5 add_config(TensorShape(8U, 8U, 2U), TensorShape(5U, 5U, 2U, 1U), TensorShape(1U), TensorShape(4U, 4U, 1U), PadStrideInfo(1, 1, 0, 0)); + add_config(TensorShape(8U, 8U, 2U), TensorShape(5U, 5U, 2U), TensorShape(1U), TensorShape(8U, 8U, 1U), PadStrideInfo(1, 1, 2, 2)); } }; -- cgit v1.2.1