commit aa06714b9f49fe36716fd3c116aadd892fadd4ac
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Tue Jun 11 16:01:44 2019 +0100
committer Áron Virginás-Tar <aron.virginas-tar@arm.com> Wed Jun 12 10:13:50 2019 +0000
tree 9c2e371d8ff914be6203046751ac18364906f418
parent 7f6db40e8c02084a5e960e1f377e8e83ed87f6a1

IVGCVSW-3260 Add serialization support for SpaceToDepth

Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Ie0d9561437ff5da8fa4db81fee1e70bd18c47034

src/armnnSerializer/Serializer.cpp

diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index 676d10c..131bb95 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -788,7 +788,17 @@
                                                const armnn::SpaceToDepthDescriptor& spaceToDepthDescriptor,
                                                const char* name)
 {
-    throw armnn::Exception("SerializerVisitor::VisitSpaceToDepthLayer is not yet implemented");
+    auto flatBufferBaseLayer  = CreateLayerBase(layer, serializer::LayerType::LayerType_SpaceToDepth);
+    auto flatBufferDescriptor =
+        CreateSpaceToDepthDescriptor(m_flatBufferBuilder,
+                                     spaceToDepthDescriptor.m_BlockSize,
+                                     GetFlatBufferDataLayout(spaceToDepthDescriptor.m_DataLayout));
+
+    auto flatBufferLayer = serializer::CreateSpaceToDepthLayer(m_flatBufferBuilder,
+                                                               flatBufferBaseLayer,
+                                                               flatBufferDescriptor);
+
+    CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_SpaceToDepthLayer);
 }
 
 // Build FlatBuffer for Splitter Layer