diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-05-28 10:04:57 +0100 |
---|---|---|
committer | Giuseppe Rossini <giuseppe.rossini@arm.com> | 2019-07-16 16:08:25 +0000 |
commit | ba27e4467dfc04e23ce9483330be062e9aaebdc5 (patch) | |
tree | 3bb9e113307f4358b6f52b399b43f0efa088fc1f /arm_compute/core | |
parent | d7ed672e4c4deecb7498581790b87bfe99fcf054 (diff) | |
download | ComputeLibrary-ba27e4467dfc04e23ce9483330be062e9aaebdc5.tar.gz |
COMPMID-2236: QUANTIZED_16BIT_LSTM operator for NEON
Change-Id: I554023508e09b790ecc1bbdada529697d6c7b616
Signed-off-by: giuros01 <giuseppe.rossini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1551
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Diffstat (limited to 'arm_compute/core')
-rw-r--r-- | arm_compute/core/NEON/kernels/NEDequantizationLayerKernel.h | 4 | ||||
-rw-r--r-- | arm_compute/core/Validate.h | 4 |
2 files changed, 4 insertions, 4 deletions
diff --git a/arm_compute/core/NEON/kernels/NEDequantizationLayerKernel.h b/arm_compute/core/NEON/kernels/NEDequantizationLayerKernel.h index 3320ba6889..f0a2a57d1a 100644 --- a/arm_compute/core/NEON/kernels/NEDequantizationLayerKernel.h +++ b/arm_compute/core/NEON/kernels/NEDequantizationLayerKernel.h @@ -52,13 +52,13 @@ public: ~NEDequantizationLayerKernel() = default; /** Set input, output tensors. * - * @param[in] input Source tensor. Data type supported: QASYMM8/QSYMM8. + * @param[in] input Source tensor. Data type supported: QASYMM8/QSYMM8/QSYMM16. * @param[out] output Destination tensor with the same dimensions of input. Data type supported: F16/F32. */ void configure(const ITensor *input, ITensor *output); /** Static function to check if given info will lead to a valid configuration of @ref NEDequantizationLayerKernel * - * @param[in] input Input tensor info. Data types supported: QASYMM8/QSYMM8. + * @param[in] input Input tensor info. Data types supported: QASYMM8/QSYMM8/QSYMM16. * @param[in] output Output tensor info. Data types supported: F16/F32. * * @return a status diff --git a/arm_compute/core/Validate.h b/arm_compute/core/Validate.h index dab4221a3b..37c7b50ec7 100644 --- a/arm_compute/core/Validate.h +++ b/arm_compute/core/Validate.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2016-2018 ARM Limited. + * Copyright (c) 2016-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -565,7 +565,7 @@ inline arm_compute::Status error_on_mismatching_quantization_info(const char *fu DataType &&first_data_type = tensor_info_1->data_type(); const QuantizationInfo first_quantization_info = tensor_info_1->quantization_info(); - if(!is_data_type_quantized_asymmetric(first_data_type)) + if(!is_data_type_quantized(first_data_type)) { return arm_compute::Status{}; } |