commit bcd3c85b5a7657b38f503676b88a80ae74165acd
author Matteo Martincigh <matteo.martincigh@arm.com> Fri Sep 28 14:14:12 2018 +0100
committer Matthew Bentham <matthew.bentham@arm.com> Wed Oct 10 16:16:57 2018 +0100
tree a1f4f934aea17f0faa30f64b9eec0108bbe09ad3
parent e1fa50c70c647e3ba6dec0eb49ce82a3820fb6bb

IVGCVSW-1863 Support NHWC for L2Normalization

 * Added L2NormalizationDescriptor struct with m_DataLyaout member
 * Updated all IsL2NormalizationSupported calls to take a descriptor
   as an argument
 * Updated L2NormalizationLayer to take a descriptor as an argument

!android-nn-driver:150116

Change-Id: I0459352d19cfd269bc864a70cf73910bf44fdc01

src/armnn/Network.cpp

diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index 4f5e297..49e60e1 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -468,9 +468,10 @@
     return m_Graph->AddLayer<ResizeBilinearLayer>(resizeDescriptor,name);
 }
 
-IConnectableLayer* Network::AddL2NormalizationLayer(const char* name)
+IConnectableLayer* Network::AddL2NormalizationLayer(const L2NormalizationDescriptor& desc,
+                                                    const char* name)
 {
-    return m_Graph->AddLayer<L2NormalizationLayer>(name);
+    return m_Graph->AddLayer<L2NormalizationLayer>(desc, name);
 }
 
 IConnectableLayer* Network::AddConstantLayer(const ConstTensor& input, const char* name)