diff options
author | Keith Davis <keith.davis@arm.com> | 2020-02-19 10:08:33 +0000 |
---|---|---|
committer | James Conroy <james.conroy@arm.com> | 2020-02-20 14:14:38 +0000 |
commit | a90573581bb2f63f229ddd61650163b9fc0fdd3e (patch) | |
tree | 80f272650985f9cd7ae684791dc54d6f68bfd1ca /src/armnnTfLiteParser/TfLiteParser.cpp | |
parent | c02d4411b6bf9d755e25a1593c9fe16e7d93901e (diff) | |
download | armnn-a90573581bb2f63f229ddd61650163b9fc0fdd3e.tar.gz |
IVGCVSW-4472 Yolo v3 ExecuteNetwork failing
* Add Debug workload for QAsymmS8/U8
* Change Dequantize tests to test AsymmS8 instead of SymmS8
* Fix incorrect supportedness within RefLayerSupport
Signed-off-by: Keith Davis <keith.davis@arm.com>
Change-Id: Ie51f1e33c564d46c86bf0150b1addda3fc093d13
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.cpp')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index 593f3eb02d..f5c01f249a 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -316,7 +316,7 @@ armnn::TensorInfo ToTensorInfo(TfLiteParser::TensorRawPtr tensorPtr, const std:: type = armnn::DataType::Float32; break; case tflite::TensorType_INT8: - if (tensorPtr->quantization->zero_point.size() == 1 && tensorPtr->quantization->zero_point[0] != 0) + if (tensorPtr->quantization->zero_point.size() == 1) { // Per-tensor type = armnn::DataType::QAsymmS8; @@ -398,7 +398,6 @@ armnn::TensorInfo ToTensorInfo(TfLiteParser::TensorRawPtr tensorPtr, const std:: quantizationScales, dimensionMappings[boost::numeric_cast<unsigned int>( tensorPtr->quantization->quantized_dimension)]); - return result; } } @@ -2900,6 +2899,11 @@ TfLiteParser::CreateConstTensor(TensorRawPtr tensorPtr, tensorPtr, tensorInfo, permutationVector); + case armnn::DataType::QAsymmS8: + return CreateConstTensorAndStoreData<int8_t>(bufferPtr, + tensorPtr, + tensorInfo, + permutationVector); case armnn::DataType::Signed32: return CreateConstTensorAndStoreData<int32_t>(bufferPtr, tensorPtr, |