diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-08-22 11:44:04 +0100 |
|---|---|---|
| committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-08-22 15:28:03 +0000 |
| commit | c8530210c17b391f27ace95523e9590e8166fcd8 (patch) | |
| tree | 7ab0ea38b730a194aba2da91440a991a7614be8f /src | |
| parent | 57a896172ff77c13655b1dc5acc9cfb2930e0570 (diff) | |
| download | ComputeLibrary-c8530210c17b391f27ace95523e9590e8166fcd8.tar.gz | |
COMPMID-2417: Add new QASYMM8_PER_CHANNEL data type
Change-Id: I6825320909a553513b98cf9b262fc90e37a2fa30
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1790
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'src')
| -rw-r--r-- | src/core/Utils.cpp | 1 | ||||
| -rw-r--r-- | src/runtime/CL/CLTensorAllocator.cpp | 2 |
2 files changed, 2 insertions, 1 deletions
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); |