diff options
| author | Pablo Tello <pablo.tello@arm.com> | 2018-01-30 11:10:13 +0000 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:45:00 +0000 |
| commit | fc9bda3d99cab79ffeab8ae329e0648a257472c2 (patch) | |
| tree | c8806f99027101ea07b9cdf059ee22d849b08291 | |
| parent | d466c2d25346ec0b4ed1cc122de93c54534107b3 (diff) | |
| download | ComputeLibrary-fc9bda3d99cab79ffeab8ae329e0648a257472c2.tar.gz | |
COMPMID-815: Fixed Winograd 5x5 padding bug.
Change-Id: I38ae204632ae27c5fe7a0131462343397899868c
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/118120
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
| -rw-r--r-- | src/core/NEON/kernels/winograd/winograd_gemm.cpp | 5 |
1 files changed, 2 insertions, 3 deletions
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<kr, kc, itr, itc>::Convolution<TOut, TIn>::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 }; |