diff options
author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2018-08-09 13:28:41 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
commit | dff601dad65d95ad85fee349a44b6417b7de7f2c (patch) | |
tree | 4770f08b020ec3b1caa1d51e46e567fbfbf1aa0c /arm_compute/runtime | |
parent | 4dc96a9c60934fe57a99836df21c5812f6257fb6 (diff) | |
download | ComputeLibrary-dff601dad65d95ad85fee349a44b6417b7de7f2c.tar.gz |
COMPMID-1488 - Add support for NHWC when running CLGEMMConvolutionLayer with QASYMM8
Fixed also a bug in the graph API related to the bias shape in DepthWiseConvolution for NHWC
Change-Id: I275141a42e51f6747b77db1c31d1bc69e8685af5
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/143454
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'arm_compute/runtime')
-rw-r--r-- | arm_compute/runtime/CL/functions/CLGEMMConvolutionLayer.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/arm_compute/runtime/CL/functions/CLGEMMConvolutionLayer.h b/arm_compute/runtime/CL/functions/CLGEMMConvolutionLayer.h index 7c272a348b..8538d83c2b 100644 --- a/arm_compute/runtime/CL/functions/CLGEMMConvolutionLayer.h +++ b/arm_compute/runtime/CL/functions/CLGEMMConvolutionLayer.h @@ -40,6 +40,7 @@ #include "arm_compute/runtime/CL/functions/CLGEMM.h" #include "arm_compute/runtime/CL/functions/CLGEMMLowpMatrixMultiplyCore.h" #include "arm_compute/runtime/CL/functions/CLGEMMLowpOutputStage.h" +#include "arm_compute/runtime/CL/functions/CLReshapeLayer.h" #include "arm_compute/runtime/IMemoryManager.h" #include <memory> @@ -88,7 +89,7 @@ private: * -# @ref CLGEMMLowpMatrixMultiplyCore (if the data type is QASYMM8) * -# @ref CLGEMMLowpQuantizeDownInt32ToUint8ScaleByFixedPoint (if the data type is QASYMM8) * -# @ref CLArithmeticAdditionKernel (if biases != nullptr and we have a 1x1 convolution with the NHWC data layout) - * -# @ref CLCol2ImKernel (if NCHW data layout) + * -# @ref CLCol2ImKernel (if NCHW data layout) or @ref CLReshapeLayer (if NHWC with QASYMM8) */ class CLGEMMConvolutionLayer : public IFunction { @@ -182,6 +183,7 @@ private: CLCol2ImKernel _col2im_kernel; CLActivationLayer _activationlayer_function; CLArithmeticAdditionKernel _add_bias_kernel; + CLReshapeLayer _reshape_layer; const ICLTensor *_original_weights; |