commit ee391d59dbe3305734de4ff7d98c27c8a5252624
author Nikhil Raj <nikhil.raj@arm.com> Thu Sep 05 17:50:44 2019 +0100
committer Nikhil Raj <nikhil.raj@arm.com> Thu Sep 05 17:50:44 2019 +0100
tree 642caa74ee548d15d7dde1ba686ac4f479545d2c
parent 02356de9da5d5480b4b9b413d1b6b33daf7ab717

IVGCVSW-3722 Add front end support for ArgMinMax

Change-Id: I31c5616bea3097f30cde68442d3222e0b0fe2235
Signed-off-by: Nikhil Raj <nikhil.raj@arm.com>

src/armnn/Network.cpp

diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index dc26a1b..6971cb8 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -1116,6 +1116,12 @@
     return m_Graph->AddLayer<ActivationLayer>(activationDescriptor, name);
 }
 
+IConnectableLayer* Network::AddArgMinMaxLayer(const ArgMinMaxDescriptor& argMinMaxDescriptor,
+                                              const char* name)
+{
+    return m_Graph->AddLayer<ArgMinMaxLayer>(argMinMaxDescriptor, name);
+}
+
 IConnectableLayer* Network::AddNormalizationLayer(const NormalizationDescriptor&
 normalizationDescriptor,
     const char* name)