From ddb1d06dbcb5dc4a89a237ac1176279669817f46 Mon Sep 17 00:00:00 2001 From: Francis Murtagh Date: Tue, 10 Mar 2020 13:51:45 +0000 Subject: MLCE-159 Add QAsymmS8 to ArmnnQuantizer * Allow per layer quantization from Fp32 to Int8 (QAsymmS8) like TfLite Signed-off-by: Francis Murtagh Change-Id: I5bbf770aa29d81af3568c15b47d2b2c18e55bb28 --- src/armnnSerializer/ArmnnSchema.fbs | 3 ++- src/armnnSerializer/SerializerUtils.cpp | 2 ++ 2 files changed, 4 insertions(+), 1 deletion(-) (limited to 'src/armnnSerializer') diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index d7565a5b9a..ca3db5d542 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -37,7 +37,8 @@ enum DataType : byte { Boolean = 4, QuantisedSymm16 = 5, // deprecated QAsymmU8 = 6, - QSymmS16 = 7 + QSymmS16 = 7, + QAsymmS8 = 8 } enum DataLayout : byte { diff --git a/src/armnnSerializer/SerializerUtils.cpp b/src/armnnSerializer/SerializerUtils.cpp index 02a5ed3872..c1847715a0 100644 --- a/src/armnnSerializer/SerializerUtils.cpp +++ b/src/armnnSerializer/SerializerUtils.cpp @@ -58,6 +58,8 @@ armnnSerializer::DataType GetFlatBufferDataType(armnn::DataType dataType) return armnnSerializer::DataType::DataType_Signed32; case armnn::DataType::QSymmS16: return armnnSerializer::DataType::DataType_QSymmS16; + case armnn::DataType::QAsymmS8: + return armnnSerializer::DataType::DataType_QAsymmS8; case armnn::DataType::QAsymmU8: return armnnSerializer::DataType::DataType_QAsymmU8; case armnn::DataType::Boolean: -- cgit v1.2.1