diff options
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); |