diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-03-27 15:56:57 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2020-03-27 15:56:57 +0000 |
commit | 1a84fe3b92aaab61d2f3b865d347459e16eb43d4 (patch) | |
tree | 140928df39ee8d46e3fb4e3ccc7883116a1ce929 /src/armnnSerializer/SerializerUtils.cpp | |
parent | 57ef0088d20dd708ff92222d244ea02f1e1e5216 (diff) | |
download | armnn-1a84fe3b92aaab61d2f3b865d347459e16eb43d4.tar.gz |
IVGCVSW-4555 ArmnnConverter (Serializer) does not support per-axis quantization params
* TensorInfo can have multiple scales and quantization dimension.
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I0ff02e3766996b6a9da6dc4e92d366bc9505c77d
Diffstat (limited to 'src/armnnSerializer/SerializerUtils.cpp')
-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 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: |