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/armnnSerializer/Serializer.hpp

diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp
index 7400885..190ed23 100644
--- a/src/armnnSerializer/Serializer.hpp
+++ b/src/armnnSerializer/Serializer.hpp
@@ -52,6 +52,10 @@
     void VisitAdditionLayer(const armnn::IConnectableLayer* layer,
                             const char* name = nullptr) override;
 
+    void VisitArgMinMaxLayer(const armnn::IConnectableLayer* layer,
+                             const armnn::ArgMinMaxDescriptor& argMinMaxDescriptor,
+                             const char* name = nullptr) override;
+
     void VisitBatchToSpaceNdLayer(const armnn::IConnectableLayer* layer,
                                   const armnn::BatchToSpaceNdDescriptor& descriptor,
                                   const char* name = nullptr) override;