From fc9bda3d99cab79ffeab8ae329e0648a257472c2 Mon Sep 17 00:00:00 2001 From: Pablo Tello Date: Tue, 30 Jan 2018 11:10:13 +0000 Subject: COMPMID-815: Fixed Winograd 5x5 padding bug. Change-Id: I38ae204632ae27c5fe7a0131462343397899868c Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/118120 Reviewed-by: Anthony Barbier Tested-by: Jenkins --- src/core/NEON/kernels/winograd/winograd_gemm.cpp | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'src/core') diff --git a/src/core/NEON/kernels/winograd/winograd_gemm.cpp b/src/core/NEON/kernels/winograd/winograd_gemm.cpp index fcfa635232..b45f6f55d9 100644 --- a/src/core/NEON/kernels/winograd/winograd_gemm.cpp +++ b/src/core/NEON/kernels/winograd/winograd_gemm.cpp @@ -34,11 +34,10 @@ Tensor4DShape WinogradGEMM::Convolution::get_output const PaddingType padding ) { - // TODO Accept different kernel sizes return Tensor4DShape { in_shape.n_batches, - (padding == PADDING_SAME) ? in_shape.n_rows : in_shape.n_rows - 2, - (padding == PADDING_SAME) ? in_shape.n_cols : in_shape.n_cols - 2, + (padding == PADDING_SAME) ? in_shape.n_rows : in_shape.n_rows - (kernel_rows - 2), + (padding == PADDING_SAME) ? in_shape.n_cols : in_shape.n_cols - (kernel_cols - 2), kernel_shape.n_output_channels, in_shape.ordering }; -- cgit v1.2.1