From 6c89ffac750010cb9335794defe8a366c04db937 Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Fri, 24 Jan 2020 12:05:05 +0000 Subject: COMPMID-2819: Fix supported types in DepthwiseReshapeKernel Add support for all types in validation of CLDepthwiseConvolutionLayerReshapeWeightsKernel. Also added a few missing parentheses in reduction_operation.cl. Change-Id: I7371ce6d5e402e9c2faafd7948e4643298bc1ace Signed-off-by: Michalis Spyrou Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2641 Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Georgios Pinitas --- .../core/CL/kernels/CLDepthwiseConvolutionLayerReshapeWeightsKernel.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'arm_compute') diff --git a/arm_compute/core/CL/kernels/CLDepthwiseConvolutionLayerReshapeWeightsKernel.h b/arm_compute/core/CL/kernels/CLDepthwiseConvolutionLayerReshapeWeightsKernel.h index f209629b11..97225c7c33 100644 --- a/arm_compute/core/CL/kernels/CLDepthwiseConvolutionLayerReshapeWeightsKernel.h +++ b/arm_compute/core/CL/kernels/CLDepthwiseConvolutionLayerReshapeWeightsKernel.h @@ -47,7 +47,7 @@ public: /** Initialize the function's source and destination. * - * @param[in] input The input tensor of dimension [IFM, W, H]. Data types supported: QASYMM8/QASYMM8_SIGNED. Data layouts supported: NHWC + * @param[in] input The input tensor of dimension [IFM, W, H]. Data types supported: All. Data layouts supported: NHWC * @param[out] output The output tensor of dimension [W*H*C0, ceil(IFM/C0)]. C0 is the number of channels read by each thread. Data types supported: same as @p weights. * @param[in] info Depthwise convolution information to reshape the input tensor. */ @@ -55,7 +55,7 @@ public: /** Static function to check if given info will lead to a valid configuration of @ref CLDepthwiseConvolutionLayer3x3NHWCKernel * - * @param[in] input The input tensor info of dimension [IFM, W, H]. Data types supported: QASYMM8/QASYMM8_SIGNED. Data layouts supported: NHWC + * @param[in] input The input tensor info of dimension [IFM, W, H]. Data types supported: All. Data layouts supported: NHWC * @param[in] output The output tensor info of dimension [W*H*C0, ceil(IFM/C0)]. C0 is the number of channels read by each thread. Data types supported: same as @p weights. * @param[in] info Depthwise convolution information to reshape the input tensor. * -- cgit v1.2.1