commit b5433ee34fd9d38c1453dc062b36348d65677002
author Matthew Jackson <matthew.jackson@arm.com> Thu Jul 11 15:54:20 2019 +0100
committer Matthew Jackson <matthew.jackson@arm.com> Tue Jul 16 09:05:14 2019 +0000
tree f48ed9f6ffb25aed908c5aa775b49bef78fd906f
parent 15a9a8f8f5e05c9967c3a52ecbfb7e173e9e61dd

IVGCVSW-3420 Add Serialization support for the new Stack layer

 * Adds serialization/deserialization support
 * Adds related unit test

Signed-off-by: Matthew Jackson <matthew.jackson@arm.com>
Change-Id: I69deb5397b8a06c679715e24971e9bb1c282140d

src/armnnSerializer/Serializer.cpp

diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index 0a9e335..b59bac60 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -929,7 +929,21 @@
                                         const armnn::StackDescriptor& stackDescriptor,
                                         const char* name)
 {
-    throw UnimplementedException("SerializerVisitor::VisitStackLayer not yet implemented");
+    auto stackBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Stack);
+
+    std::vector<unsigned int> inputShape;
+    for (unsigned int i =0; i < stackDescriptor.m_InputShape.GetNumDimensions(); i++)
+    {
+        inputShape.push_back(stackDescriptor.m_InputShape[i]);
+    }
+
+    auto flatBufferStackDescriptor = CreateStackDescriptor(m_flatBufferBuilder,
+                                                           stackDescriptor.m_Axis,
+                                                           stackDescriptor.m_NumInputs,
+                                                           m_flatBufferBuilder.CreateVector(inputShape));
+
+    auto stackLayer = serializer::CreateStackLayer(m_flatBufferBuilder, stackBaseLayer, flatBufferStackDescriptor);
+    CreateAnyLayer(stackLayer.o, serializer::Layer::Layer_StackLayer);
 }
 
 void SerializerVisitor::VisitStridedSliceLayer(const armnn::IConnectableLayer* layer,