diff options
Diffstat (limited to 'arm_compute')
-rw-r--r-- | arm_compute/core/CL/kernels/CLDepthConvertKernel.h | 7 | ||||
-rw-r--r-- | arm_compute/runtime/CL/functions/CLDepthConvert.h | 8 |
2 files changed, 11 insertions, 4 deletions
diff --git a/arm_compute/core/CL/kernels/CLDepthConvertKernel.h b/arm_compute/core/CL/kernels/CLDepthConvertKernel.h index 2c3b1b8b69..b03b90b275 100644 --- a/arm_compute/core/CL/kernels/CLDepthConvertKernel.h +++ b/arm_compute/core/CL/kernels/CLDepthConvertKernel.h @@ -43,14 +43,17 @@ public: * * Valid conversions Input -> Output : * + * - QS8 -> F32 + * - QS16 -> F32 * - U8 -> U16, S16, U32, S32 * - U16 -> U8, U32, S32 * - S16 -> U8, U32, S32 * - U32 -> U8, U16, S16 * - S32 -> U8, U16, S16 + * - F32 -> QS8, QS16 * - * @param[in] input The input tensor to convert. Data types supported: U8, U16, S16, U32 or S32. - * @param[out] output The output tensor. Data types supported: U8, U16, S16, U32 or S32. + * @param[in] input The input tensor to convert. Data types supported: U8/QS8/U16/S16/QS16/U32/S32/F32. + * @param[out] output The output tensor. Data types supported: U8/QS8/U16/S16/QS16/U32/S32/F32. * @param[in] policy Conversion policy * @param[in] shift Value for down/up conversions. Must be 0 <= shift < 8. */ diff --git a/arm_compute/runtime/CL/functions/CLDepthConvert.h b/arm_compute/runtime/CL/functions/CLDepthConvert.h index f11027656d..9a4c63dd6d 100644 --- a/arm_compute/runtime/CL/functions/CLDepthConvert.h +++ b/arm_compute/runtime/CL/functions/CLDepthConvert.h @@ -43,16 +43,20 @@ public: * * Valid conversions Input -> Output : * + * - QS8 -> F32 + * - QS16 -> F32 * - U8 -> U16, S16, U32, S32 * - U16 -> U8, U32, S32 * - S16 -> U8, U32, S32 * - U32 -> U8, U16, S16 * - S32 -> U8, U16, S16 + * - F32 -> QS8, QS16 * - * @param[in] input The input tensor to convert. Data types supported: U8, U16, S16, U32 or S32. - * @param[out] output The output tensor. Data types supported: U8, U16, S16, U32 or S32. + * @param[in] input The input tensor to convert. Data types supported: QS8/U8/U16/S16/Q16/U32/S32/F32. + * @param[out] output The output tensor. Data types supported: QS8/U8/U16/S16/QS16/U32/S32/F32. * @param[in] policy Conversion policy. * @param[in] shift Value for down/up conversions. Must be 0 <= shift < 8. + * It is not used on fixed point conversion. */ void configure(const ICLTensor *input, ICLTensor *output, ConvertPolicy policy, uint32_t shift); }; |