diff options
Diffstat (limited to 'delegate/src/DelegateUtils.hpp')
-rw-r--r-- | delegate/src/DelegateUtils.hpp | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/delegate/src/DelegateUtils.hpp b/delegate/src/DelegateUtils.hpp index f12aee9d2b..729a8b4e98 100644 --- a/delegate/src/DelegateUtils.hpp +++ b/delegate/src/DelegateUtils.hpp @@ -317,7 +317,14 @@ armnn::TensorInfo GetTensorInfoForTfLiteTensor(const TfLiteTensor& tfLiteTensor) type = armnn::DataType::QAsymmU8; break; case kTfLiteInt8: - type = armnn::DataType::QSymmS8; + if (tfLiteTensor.params.zero_point == 0) + { + type = armnn::DataType::QSymmS8; + } + else + { + type = armnn::DataType::QAsymmS8; + } break; case kTfLiteInt16: type = armnn::DataType::QSymmS16; |