commit f81edaacc38d6edc4a2dc230460120c6f83e0cda
author Matteo Martincigh <matteo.martincigh@arm.com> Mon Mar 04 14:34:30 2019 +0000
committer Matteo Martincigh <matteo.martincigh@arm.com> Mon Mar 04 16:06:42 2019 +0000
tree f2d1833efde82b1254c96b343810c56e80d7eb40
parent 5c7ab05555d3abd23a11db403090b7563ce388e6

IVGCVSW-2691 Add Serialize/Deseralize Gather layer

Change-Id: I445c4475e5abfe500b61ce8b7138c45322043c8b
Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>

src/armnnSerializer/Serializer.cpp

diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index 38e815d..4e408ae 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -331,10 +331,8 @@
 
 void SerializerVisitor::VisitGatherLayer(const armnn::IConnectableLayer* layer, const char* name)
 {
-    auto fbBaseLayer  = CreateLayerBase(layer, serializer::LayerType::LayerType_Gather);
-
-    auto flatBufferLayer = CreateGatherLayer(m_flatBufferBuilder,
-                                             fbBaseLayer);
+    auto fbGatherBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Gather);
+    auto flatBufferLayer   = serializer::CreateGatherLayer(m_flatBufferBuilder, fbGatherBaseLayer);
 
     CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_GatherLayer);
 }
@@ -342,7 +340,7 @@
 void SerializerVisitor::VisitGreaterLayer(const armnn::IConnectableLayer* layer, const char* name)
 {
     auto fbGreaterBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Greater);
-    auto fbGreaterLayer = serializer::CreateGreaterLayer(m_flatBufferBuilder, fbGreaterBaseLayer);
+    auto fbGreaterLayer     = serializer::CreateGreaterLayer(m_flatBufferBuilder, fbGreaterBaseLayer);
 
     CreateAnyLayer(fbGreaterLayer.o, serializer::Layer::Layer_GreaterLayer);
 }