commit ae42f01cb7f75fd5f7681f142f073f6f51e2e57f
author Mike Kelly <mike.kelly@arm.com> Fri Apr 24 15:44:01 2020 +0100
committer mike.kelly <mike.kelly@arm.com> Fri Apr 24 15:44:54 2020 +0000
tree 9afb538ad8e2ae1e82c0a7152515b9610951522c
parent ec52c3a04134e286d43cd2d22b916302f784ad29

Added new DataTypes to Serializer

 * Added support for BFloat16 and QAsymmS8 to Serializr.

Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I481af26d5758190ec2e5cd9f9d232d72bcb7acbd

src/armnnSerializer/Serializer.cpp

diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index cb7a5c4..3556736 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -1472,14 +1472,7 @@
             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 @@
             break;
         }
         case armnn::DataType::QSymmS8:
+        case armnn::DataType::QAsymmS8:
         case armnn::DataType::QAsymmU8:
         case armnn::DataType::Boolean:
         default: