diff options
| author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-03-19 12:26:36 +0000 |
|---|---|---|
| committer | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-03-19 12:26:36 +0000 |
| commit | cd5ac23039dc3acd750c44442960280cc33307ef (patch) | |
| tree | 37de0bae199345355535448b6c1ce2c624a298b8 /src/armnnSerializer | |
| parent | 2a326b5bbff91ffaedd9017f388aa45862047743 (diff) | |
| download | armnn-cd5ac23039dc3acd750c44442960280cc33307ef.tar.gz | |
IVGCVSW-2839 Add QuantisedSymm16 support to the ArmNN frontend
Change-Id: I76f4e7db55b0efbb1a5f7d1a32bf451e66add0c7
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Diffstat (limited to 'src/armnnSerializer')
| -rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 3 | ||||
| -rw-r--r-- | src/armnnSerializer/Serializer.cpp | 9 |
2 files changed, 11 insertions, 1 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index 00c1a4502b..a11eeadf12 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -27,7 +27,8 @@ enum DataType : byte { Float32 = 1, QuantisedAsymm8 = 2, Signed32 = 3, - Boolean = 4 + Boolean = 4, + QuantisedSymm16 = 5 } enum DataLayout : byte { diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index 0340f56d17..cf67022267 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -855,6 +855,15 @@ flatbuffers::Offset<serializer::ConstTensor> fbPayload = flatBuffersData.o; break; } + case armnn::DataType::QuantisedSymm16: + { + auto fbVector = CreateDataVector<int16_t>(constTensor.GetMemoryArea(), constTensor.GetNumBytes()); + flatbuffers::Offset<serializer::ShortData> flatBuffersData = serializer::CreateShortData( + m_flatBufferBuilder, + fbVector); + fbPayload = flatBuffersData.o; + break; + } case armnn::DataType::QuantisedAsymm8: case armnn::DataType::Boolean: default: |