diff options
| author | Pablo Tello <pablo.tello@arm.com> | 2018-01-30 14:48:11 +0000 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:45:00 +0000 |
| commit | 52140b42f4f663da7f4537abbdebd13df541bcea (patch) | |
| tree | 16c7e4b8969830fcb65860cdffdcc06c2265180c /src/core/NEON/kernels/winograd/winograd_gemm.cpp | |
| parent | 054a7144cf9c9cf7ed25adcb7e8095b9bcf866bf (diff) | |
| download | ComputeLibrary-52140b42f4f663da7f4537abbdebd13df541bcea.tar.gz | |
COMPMID-784: Winograd tramsforms refactoring
1) Removed the example files winograd_layer.hpp/cpp
2) Teplatized winograd transform kernels
Change-Id: I7045fa0b801b9d30a11275914aaa2dafd254aed2
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/118332
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/winograd/winograd_gemm.cpp')
| -rw-r--r-- | src/core/NEON/kernels/winograd/winograd_gemm.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/core/NEON/kernels/winograd/winograd_gemm.cpp b/src/core/NEON/kernels/winograd/winograd_gemm.cpp index b45f6f55d9..05426450a6 100644 --- a/src/core/NEON/kernels/winograd/winograd_gemm.cpp +++ b/src/core/NEON/kernels/winograd/winograd_gemm.cpp @@ -36,8 +36,8 @@ Tensor4DShape WinogradGEMM<kr, kc, itr, itc>::Convolution<TOut, TIn>::get_output { return Tensor4DShape { in_shape.n_batches, - (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), + (padding == PADDING_SAME) ? in_shape.n_rows : in_shape.n_rows - (kernel_rows - 1), + (padding == PADDING_SAME) ? in_shape.n_cols : in_shape.n_cols - (kernel_cols - 1), kernel_shape.n_output_channels, in_shape.ordering }; |