diff options
Diffstat (limited to 'delegate/src/DelegateUtils.hpp')
| -rw-r--r-- | delegate/src/DelegateUtils.hpp | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/delegate/src/DelegateUtils.hpp b/delegate/src/DelegateUtils.hpp index 0537ba911b..fad07ff267 100644 --- a/delegate/src/DelegateUtils.hpp +++ b/delegate/src/DelegateUtils.hpp @@ -342,14 +342,26 @@ armnn::DataType GetDataType(const TfLiteTensor& tfLiteTensor) case kTfLiteUInt8: return armnn::DataType::QAsymmU8; case kTfLiteInt8: - if (tfLiteTensor.params.zero_point == 0) + { + auto quantizationInfo = tfLiteTensor.quantization; + if (quantizationInfo.type == kTfLiteAffineQuantization) { - return armnn::DataType::QSymmS8; + auto* quantization = + reinterpret_cast<TfLiteAffineQuantization*>(tfLiteTensor.quantization.params); + if (quantization->zero_point != nullptr && quantization->zero_point->size == 1) + { + return armnn::DataType::QAsymmS8; + } + else + { + return armnn::DataType::QSymmS8; + } } else { return armnn::DataType::QAsymmS8; } + } case kTfLiteInt16: return armnn::DataType::QSymmS16; case kTfLiteInt32: |