From 7a0212aae4fe6617eb31c734c24246c345556391 Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Tue, 14 Apr 2020 16:08:32 +0100 Subject: COMPMID-3236: Add support QSYMM16 into S32 CLPixelwiseMultiplicationKernel Change-Id: Ifc519f53f04fcb14ddb9c17f98cc687f34285c97 Signed-off-by: Michele Di Giorgio Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3018 Tested-by: Arm Jenkins Reviewed-by: Michalis Spyrou Comments-Addressed: Arm Jenkins --- .../runtime/CL/functions/CLPixelWiseMultiplication.h | 20 ++++++++++++++++++-- 1 file changed, 18 insertions(+), 2 deletions(-) (limited to 'arm_compute/runtime/CL/functions/CLPixelWiseMultiplication.h') diff --git a/arm_compute/runtime/CL/functions/CLPixelWiseMultiplication.h b/arm_compute/runtime/CL/functions/CLPixelWiseMultiplication.h index a5ab829c83..47bb2bf4db 100644 --- a/arm_compute/runtime/CL/functions/CLPixelWiseMultiplication.h +++ b/arm_compute/runtime/CL/functions/CLPixelWiseMultiplication.h @@ -42,7 +42,15 @@ public: * The input tensor is [in, out] because its TensorInfo might be modified inside the kernel in case of broadcasting of dimension 0. * @param[in, out] input2 An input tensor. Data types supported: same as @p input1. * The input tensor is [in, out] because its TensorInfo might be modified inside the kernel in case of broadcasting of dimension 0. - * @param[out] output The output tensor, Data types supported: same as @p input1. Note: U8 requires both inputs to be U8. + * @param[out] output The output tensor. Data types supported: + * - U8, only if both input are U8 + * - QASYMM8, only if both inputs are QASYMM8 + * - QASYMM8_SIGNED, only if both inputs are QASYMM8_SIGNED + * - S16 + * - QSYMM16, only if both inputs are QSYMM16 + * - S32, only if both inputs are QSYMM16 + * - F16 + * - F32 * @param[in] scale Scale to apply after multiplication. * Scale must be positive and its value must be either 1/255 or 1/2^n where n is between 0 and 15. * @param[in] overflow_policy Overflow policy. Supported overflow policies: Wrap, Saturate @@ -55,7 +63,15 @@ public: * * @param[in] input1 An input tensor info. Data types supported: U8/QASYMM8/QASYMM8_SIGNED/S16/QSYMM16/F16/F32. * @param[in] input2 An input tensor info. Data types supported: same as @p input1. - * @param[in] output The output tensor info, Data types supported: same as @p input1. Note: U8 requires both inputs to be U8. + * @param[in] output The output tensor info. Data types supported: + * - U8, only if both input are U8 + * - QASYMM8, only if both inputs are QASYMM8 + * - QASYMM8_SIGNED, only if both inputs are QASYMM8_SIGNED + * - S16 + * - QSYMM16, only if both inputs are QSYMM16 + * - S32, only if both inputs are QSYMM16 + * - F16 + * - F32 * @param[in] scale Scale to apply after multiplication. * Scale must be positive and its value must be either 1/255 or 1/2^n where n is between 0 and 15. * @param[in] overflow_policy Overflow policy. Supported overflow policies: Wrap, Saturate -- cgit v1.2.1