diff options
| -rw-r--r-- | src/armnnSerializer/Serializer.cpp | 10 |
1 files changed, 2 insertions, 8 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index cb7a5c456e..355673697f 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -1472,14 +1472,7 @@ flatbuffers::Offset<serializer::ConstTensor> break; } case armnn::DataType::Float16: - { - 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::BFloat16: case armnn::DataType::QSymmS16: { auto fbVector = CreateDataVector<int16_t>(constTensor.GetMemoryArea(), constTensor.GetNumBytes()); @@ -1490,6 +1483,7 @@ flatbuffers::Offset<serializer::ConstTensor> break; } case armnn::DataType::QSymmS8: + case armnn::DataType::QAsymmS8: case armnn::DataType::QAsymmU8: case armnn::DataType::Boolean: default: |