From 1a84fe3b92aaab61d2f3b865d347459e16eb43d4 Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Fri, 27 Mar 2020 15:56:57 +0000 Subject: IVGCVSW-4555 ArmnnConverter (Serializer) does not support per-axis quantization params * TensorInfo can have multiple scales and quantization dimension. Signed-off-by: Sadik Armagan Change-Id: I0ff02e3766996b6a9da6dc4e92d366bc9505c77d --- src/armnnSerializer/SerializerUtils.cpp | 3 +++ 1 file changed, 3 insertions(+) (limited to 'src/armnnSerializer/SerializerUtils.cpp') diff --git a/src/armnnSerializer/SerializerUtils.cpp b/src/armnnSerializer/SerializerUtils.cpp index c1847715a0..5566abf7e4 100644 --- a/src/armnnSerializer/SerializerUtils.cpp +++ b/src/armnnSerializer/SerializerUtils.cpp @@ -39,6 +39,7 @@ armnnSerializer::ConstTensorData GetFlatBufferConstTensorData(armnn::DataType da case armnn::DataType::QSymmS16: return armnnSerializer::ConstTensorData::ConstTensorData_ShortData; case armnn::DataType::QAsymmU8: + case armnn::DataType::QSymmS8: case armnn::DataType::Boolean: return armnnSerializer::ConstTensorData::ConstTensorData_ByteData; default: @@ -62,6 +63,8 @@ armnnSerializer::DataType GetFlatBufferDataType(armnn::DataType dataType) return armnnSerializer::DataType::DataType_QAsymmS8; case armnn::DataType::QAsymmU8: return armnnSerializer::DataType::DataType_QAsymmU8; + case armnn::DataType::QSymmS8: + return armnnSerializer::DataType::DataType_QSymmS8; case armnn::DataType::Boolean: return armnnSerializer::DataType::DataType_Boolean; default: -- cgit v1.2.1