From c8530210c17b391f27ace95523e9590e8166fcd8 Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Thu, 22 Aug 2019 11:44:04 +0100 Subject: COMPMID-2417: Add new QASYMM8_PER_CHANNEL data type Change-Id: I6825320909a553513b98cf9b262fc90e37a2fa30 Signed-off-by: Michalis Spyrou Reviewed-on: https://review.mlplatform.org/c/1790 Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Georgios Pinitas --- src/core/Utils.cpp | 1 + src/runtime/CL/CLTensorAllocator.cpp | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) (limited to 'src') diff --git a/src/core/Utils.cpp b/src/core/Utils.cpp index d0bffdf660..0c7eea84e3 100644 --- a/src/core/Utils.cpp +++ b/src/core/Utils.cpp @@ -160,6 +160,7 @@ const std::string &arm_compute::string_from_data_type(DataType dt) { DataType::SIZET, "SIZET" }, { DataType::QSYMM8, "QSYMM8" }, { DataType::QSYMM8_PER_CHANNEL, "QSYMM8_PER_CHANNEL" }, + { DataType::QASYMM8_PER_CHANNEL, "QASYMM8_PER_CHANNEL" }, { DataType::QASYMM8, "QASYMM8" }, { DataType::QSYMM16, "QSYMM16" }, }; diff --git a/src/runtime/CL/CLTensorAllocator.cpp b/src/runtime/CL/CLTensorAllocator.cpp index f3f16cd8c0..028a764fc2 100644 --- a/src/runtime/CL/CLTensorAllocator.cpp +++ b/src/runtime/CL/CLTensorAllocator.cpp @@ -139,7 +139,7 @@ void CLTensorAllocator::allocate() } // Allocate and fill the quantization parameter arrays - if(info().data_type() == DataType::QSYMM8_PER_CHANNEL) + if(is_data_type_quantized_per_channel(info().data_type())) { const size_t pad_size = 0; populate_quantization_info(_scale, _offset, info().quantization_info(), pad_size); -- cgit v1.2.1