From ae42f01cb7f75fd5f7681f142f073f6f51e2e57f Mon Sep 17 00:00:00 2001
From: Mike Kelly <mike.kelly@arm.com>
Date: Fri, 24 Apr 2020 15:44:01 +0100
Subject: 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 | 10 ++--------
 1 file 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:
-- 
cgit v1.2.1