From 35ea9a7521b4b858a6bcc33a244b6e7f94b6e3a3 Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Fri, 23 Aug 2019 12:02:06 +0100 Subject: COMPMID-2317: Add new QASYMM16 data type Change-Id: I9fbbba633f10d542474a08b1898150b9799b7ae5 Signed-off-by: Michele Di Giorgio Reviewed-on: https://review.mlplatform.org/c/1805 Comments-Addressed: Arm Jenkins Reviewed-by: Michalis Spyrou Tested-by: Arm Jenkins --- src/core/Utils.cpp | 1 + src/runtime/CL/functions/CLLSTMLayerQuantized.cpp | 4 ++-- src/runtime/NEON/functions/NELSTMLayerQuantized.cpp | 2 +- 3 files changed, 4 insertions(+), 3 deletions(-) (limited to 'src') diff --git a/src/core/Utils.cpp b/src/core/Utils.cpp index d5d9d10e6b..122373f5c6 100644 --- a/src/core/Utils.cpp +++ b/src/core/Utils.cpp @@ -163,6 +163,7 @@ const std::string &arm_compute::string_from_data_type(DataType dt) { DataType::QASYMM8_PER_CHANNEL, "QASYMM8_PER_CHANNEL" }, { DataType::QASYMM8, "QASYMM8" }, { DataType::QSYMM16, "QSYMM16" }, + { DataType::QASYMM16, "QASYMM16" }, }; return dt_map[dt]; diff --git a/src/runtime/CL/functions/CLLSTMLayerQuantized.cpp b/src/runtime/CL/functions/CLLSTMLayerQuantized.cpp index e0006a77d0..da19cc1cc3 100644 --- a/src/runtime/CL/functions/CLLSTMLayerQuantized.cpp +++ b/src/runtime/CL/functions/CLLSTMLayerQuantized.cpp @@ -345,7 +345,7 @@ void CLLSTMLayerQuantized::run() _tanh_output_state.run(); _mul_output_state_tmp_output_gate.run(); - // Requantize output state from QSYMM16 to QASYMM16 + // Requantize output state from QSYMM16 to QASYMM8 _dequantize.run(); _quantize.run(); } @@ -394,4 +394,4 @@ void CLLSTMLayerQuantized::prepare() } } -} // namespace arm_compute \ No newline at end of file +} // namespace arm_compute diff --git a/src/runtime/NEON/functions/NELSTMLayerQuantized.cpp b/src/runtime/NEON/functions/NELSTMLayerQuantized.cpp index 6cfa9887ff..96c776141a 100644 --- a/src/runtime/NEON/functions/NELSTMLayerQuantized.cpp +++ b/src/runtime/NEON/functions/NELSTMLayerQuantized.cpp @@ -324,7 +324,7 @@ void NELSTMLayerQuantized::run() _tanh_output_state.run(); _mul3.run(); - // Requantize output state from QSYMM16 to QASYMM16 + // Requantize output state from QSYMM16 to QASYMM8 _dequantize.run(); _quantize.run(); } -- cgit v1.2.1