commit 2b8c1da565871b3e69567c2cfc46c8dcbef301aa
author Matthew Jackson <matthew.jackson@arm.com> Thu Jul 04 14:59:16 2019 +0100
committer Áron Virginás-Tar <aron.virginas-tar@arm.com> Wed Jul 10 12:06:51 2019 +0000
tree 682327de212e273405cb257028568db997644c35
parent ad5293a86e315049de36afd723dcd1a7e70681a7

IVGCVSW-3418 Add Arm NN front end support for the new Stack layer

 * Added new StackLayer class
 * Made necessary changes to Descriptors, ILayerSupport, ILayerVisitor, etc.
 * Added unit tests

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

src/armnn/Network.cpp

diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index 3b7a1cf..2949381 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -1422,6 +1422,12 @@
     return layer;
 }
 
+IConnectableLayer* Network::AddStackLayer(const StackDescriptor& stackDescriptor,
+                                          const char* name)
+{
+    return m_Graph->AddLayer<StackLayer>(stackDescriptor, name);
+}
+
 void Network::Accept(ILayerVisitor& visitor) const
 {
     for (auto layer : GetGraph())