diff options
author | Tee Jung <naey05@gmail.com> | 2019-10-23 06:42:24 +0000 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2019-10-26 21:50:36 +0100 |
commit | c34e53b469e2ca66a4978ff80d67333e4fef9ba0 (patch) | |
tree | d88562b73f6ba052e1b080108dd86b7ee6c61055 /src | |
parent | 830101c2debac8b3f22c048fad603f34b76bdca1 (diff) | |
download | armnn-c34e53b469e2ca66a4978ff80d67333e4fef9ba0.tar.gz |
Bug fix on serialization of tensor with qsymm16 type
Signed-off-by: Jung Tae-young <naey05@gmail.com>
Change-Id: Ie3958c25fae752a8ec010ff6c77e4b625dec1f68
Diffstat (limited to 'src')
-rw-r--r-- | src/armnnSerializer/SerializerUtils.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/armnnSerializer/SerializerUtils.cpp b/src/armnnSerializer/SerializerUtils.cpp index 719f8a92dd..908da6450c 100644 --- a/src/armnnSerializer/SerializerUtils.cpp +++ b/src/armnnSerializer/SerializerUtils.cpp @@ -36,6 +36,7 @@ armnnSerializer::ConstTensorData GetFlatBufferConstTensorData(armnn::DataType da case armnn::DataType::Signed32: return armnnSerializer::ConstTensorData::ConstTensorData_IntData; case armnn::DataType::Float16: + case armnn::DataType::QuantisedSymm16: return armnnSerializer::ConstTensorData::ConstTensorData_ShortData; case armnn::DataType::QuantisedAsymm8: case armnn::DataType::Boolean: @@ -55,6 +56,8 @@ armnnSerializer::DataType GetFlatBufferDataType(armnn::DataType dataType) return armnnSerializer::DataType::DataType_Float16; case armnn::DataType::Signed32: return armnnSerializer::DataType::DataType_Signed32; + case armnn::DataType::QuantisedSymm16: + return armnnSerializer::DataType::DataType_QuantisedSymm16; case armnn::DataType::QuantisedAsymm8: return armnnSerializer::DataType::DataType_QuantisedAsymm8; case armnn::DataType::Boolean: |