diff options
author | Manuel Bottini <manuel.bottini@arm.com> | 2019-06-14 17:18:12 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-06-20 17:41:16 +0000 |
commit | 3689fcd5915cd902cb4ea5f618f2a6e42f6dc4a1 (patch) | |
tree | 9d068e7548af2a6054468c35f3259b52bdda91a5 /arm_compute/core/QuantizationInfo.h | |
parent | a28aebc5f80630683f50ce934e8584c6b0004beb (diff) | |
download | ComputeLibrary-3689fcd5915cd902cb4ea5f618f2a6e42f6dc4a1.tar.gz |
COMPMID-2408: Add QSYMM16 support for ElementwiseAddition for NEON
Change-Id: I22991e9369ffba9b51a94522ff4977933e887b94
Signed-off-by: Manuel Bottini <manuel.bottini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1352
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Giuseppe Rossini <giuseppe.rossini@arm.com>
Diffstat (limited to 'arm_compute/core/QuantizationInfo.h')
-rw-r--r-- | arm_compute/core/QuantizationInfo.h | 53 |
1 files changed, 53 insertions, 0 deletions
diff --git a/arm_compute/core/QuantizationInfo.h b/arm_compute/core/QuantizationInfo.h index 94f7e76c3e..06c9b61154 100644 --- a/arm_compute/core/QuantizationInfo.h +++ b/arm_compute/core/QuantizationInfo.h @@ -25,6 +25,7 @@ #define __ARM_COMPUTE_QUANTIZATION_INFO_H__ #include "arm_compute/core/Rounding.h" +#include "utils/misc/Utility.h" #include <cstddef> #include <vector> @@ -255,5 +256,57 @@ inline float dequantize_qsymm8(int8_t value, const QuantizationInfo &qinfo) { return value * qinfo.uniform().scale; } + +/** Quantize a value given a 16-bit symmetric quantization scheme + * + * @param[in] value Value to quantize + * @param[in] qinfo Quantization information to use for quantizing + * @param[in] rounding_policy (Optional) Rounding policy to use. Default: nearest up + * + * @return Quantized value + */ +inline int16_t quantize_qsymm16(float value, const UniformQuantizationInfo &qinfo, RoundingPolicy rounding_policy = RoundingPolicy::TO_NEAREST_UP) +{ + int quantized = arm_compute::round(value / qinfo.scale, rounding_policy); + quantized = arm_compute::utility::clamp<int, int16_t>(quantized); + return quantized; +} + +/** Dequantize a value given a 16-bit symmetric quantization scheme + * + * @param[in] value Value to dequantize + * @param[in] qinfo Quantization information to use for dequantizing + * + * @return Dequantized value + */ +inline float dequantize_qsymm16(int16_t value, const UniformQuantizationInfo &qinfo) +{ + return value * qinfo.scale; +} + +/** Quantize a value given a 16-bit symmetric quantization scheme + * + * @param[in] value Value to quantize + * @param[in] qinfo Quantization information to use for quantizing + * + * @return Quantized value + */ +inline int16_t quantize_qsymm16(float value, const QuantizationInfo &qinfo) +{ + return quantize_qsymm16(value, qinfo.uniform()); +} + +/** Dequantize a value given a 16-bit symmetric quantization scheme + * + * @param[in] value Value to dequantize + * @param[in] qinfo Quantization information to use for dequantizing + * + * @return Dequantized value + */ +inline float dequantize_qsymm16(int16_t value, const QuantizationInfo &qinfo) +{ + return dequantize_qsymm16(value, qinfo.uniform()); +} + } // namespace arm_compute #endif /*__ARM_COMPUTE_QUANTIZATION_INFO_H__ */
\ No newline at end of file |