| commit | d4b2c9f702bf6631e0aa99deefd66c60621850c3 | [log] [tgz] |
|---|---|---|
| author | Michele Di Giorgio <michele.digiorgio@arm.com> | Thu May 10 10:49:32 2018 +0100 |
| committer | Anthony Barbier <anthony.barbier@arm.com> | Fri Nov 02 16:52:54 2018 +0000 |
| tree | 4b7192fbcd92d2e642119964d36a3fcbd3e6164d | |
| parent | e8c18d48f47e309f5a5b1d983614530dc17b28be [diff] [blame] |
COMPMID-1270: Allow having weights with 1 IFM and multiple OFMs in CLDepthwiseConvolution3x3NCHW Change-Id: Ib2526f18bf303afd498ff85ca18c8df876f545ed Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/134546 Tested-by: Jenkins <bsgcomp@arm.com> Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
diff --git a/src/core/CL/kernels/CLDepthwiseConvolutionLayer3x3NCHWKernel.cpp b/src/core/CL/kernels/CLDepthwiseConvolutionLayer3x3NCHWKernel.cpp index ca18957..d5b34c3 100644 --- a/src/core/CL/kernels/CLDepthwiseConvolutionLayer3x3NCHWKernel.cpp +++ b/src/core/CL/kernels/CLDepthwiseConvolutionLayer3x3NCHWKernel.cpp
@@ -68,7 +68,7 @@ { ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(weights, biases); } - ARM_COMPUTE_RETURN_ERROR_ON(biases->dimension(0) != weights->dimension(2)); + ARM_COMPUTE_RETURN_ERROR_ON((biases->dimension(0) != weights->dimension(2)) && (weights->dimension(2) != 1 || biases->dimension(0) != weights->dimension(3))); ARM_COMPUTE_RETURN_ERROR_ON(biases->num_dimensions() > 1); }