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/armnn/Network.cpp

diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index 6bd365b..3e7d4d5 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -835,6 +835,12 @@
     return m_Graph->AddLayer<SpaceToBatchNdLayer>(spaceToBatchNdDescriptor, name);
 }
 
+IConnectableLayer* Network::AddSpaceToDepthLayer(const SpaceToDepthDescriptor& spaceToDepthDescriptor,
+                                                 const char* name)
+{
+    return m_Graph->AddLayer<SpaceToDepthLayer>(spaceToDepthDescriptor, name);
+}
+
 IConnectableLayer* Network::AddFloorLayer(const char* name)
 {
     return m_Graph->AddLayer<FloorLayer>(name);