commit 972af1568372f243f241b58fa07f0d3a7f991d1e
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Tue Jun 11 14:14:03 2019 +0100
committer Áron Virginás-Tar <aron.virginas-tar@arm.com> Wed Jun 12 08:53:33 2019 +0000
tree 7f00157cc66bef8ced30a8e0a0f5e5305a82111f
parent e9e1878cb94e45ba71303a8ceda5bc523997fde4

IVGCVSW-3258 Add front end support for new SpaceToDepth layer

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

src/armnnSerializer/Serializer.cpp

diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index ccf2199..676d10c 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -783,6 +783,14 @@
     CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_SpaceToBatchNdLayer);
 }
 
+// Build FlatBuffer for SpaceToDepthLayer
+void SerializerVisitor::VisitSpaceToDepthLayer(const armnn::IConnectableLayer* layer,
+                                               const armnn::SpaceToDepthDescriptor& spaceToDepthDescriptor,
+                                               const char* name)
+{
+    throw armnn::Exception("SerializerVisitor::VisitSpaceToDepthLayer is not yet implemented");
+}
+
 // Build FlatBuffer for Splitter Layer
 void SerializerVisitor::VisitSplitterLayer(const armnn::IConnectableLayer* layer,
                                            const armnn::ViewsDescriptor& viewsDescriptor,

src/armnnSerializer/Serializer.hpp

diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp
index 2e2816a..c9416bb 100644
--- a/src/armnnSerializer/Serializer.hpp
+++ b/src/armnnSerializer/Serializer.hpp
@@ -182,6 +182,10 @@
                                   const armnn::SpaceToBatchNdDescriptor& spaceToBatchNdDescriptor,
                                   const char* name = nullptr) override;
 
+    void VisitSpaceToDepthLayer(const armnn::IConnectableLayer* layer,
+                                const armnn::SpaceToDepthDescriptor& spaceToDepthDescriptor,
+                                const char* name = nullptr) override;
+
     void VisitNormalizationLayer(const armnn::IConnectableLayer* layer,
                                  const armnn::NormalizationDescriptor& normalizationDescriptor,
                                  const char* name = nullptr) override;