aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/CL/functions/CLWinogradConvolutionLayer.cpp
diff options
context:
space:
mode:
authorGian Marco Iodice <gianmarco.iodice@arm.com>2018-07-03 12:22:09 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:54:10 +0000
commit876be2a0d11874d871860dbd22481f831d6878f6 (patch)
tree907afac009513f882b39b2770b187635cebd1664 /src/runtime/CL/functions/CLWinogradConvolutionLayer.cpp
parent09daf4ddf5940d18ce95e7dd0859d1dace3b133e (diff)
downloadComputeLibrary-876be2a0d11874d871860dbd22481f831d6878f6.tar.gz
COMPMID-1339 - Implementing Winograd Convolution Layer 1x5 and 5x1 kernels on OpenCL NCHW
Change-Id: Ia293cd89651146a0e27e5f7c74ca9c924807e83c Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/138707 Tested-by: Jenkins <bsgcomp@arm.com> Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'src/runtime/CL/functions/CLWinogradConvolutionLayer.cpp')
-rw-r--r--src/runtime/CL/functions/CLWinogradConvolutionLayer.cpp3
1 files changed, 2 insertions, 1 deletions
diff --git a/src/runtime/CL/functions/CLWinogradConvolutionLayer.cpp b/src/runtime/CL/functions/CLWinogradConvolutionLayer.cpp
index 11714fac41..f9ea91ddc4 100644
--- a/src/runtime/CL/functions/CLWinogradConvolutionLayer.cpp
+++ b/src/runtime/CL/functions/CLWinogradConvolutionLayer.cpp
@@ -59,7 +59,8 @@ Size2D winograd_output_tile(const Size2D &input_dims, const Size2D &kernel_dims)
}
else if(kernel_max_dim == 5U)
{
- output_tile = Size2D(4U, 4U);
+ output_tile = Size2D(kernel_dims.width == 1 ? 1U : 4U,
+ kernel_dims.height == 1 ? 1U : 4U);
}
return output_tile;