commit fc413c0c977e6c9680a2aa6546e977be0a2efdb9
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Wed Feb 13 15:41:52 2019 +0000
committer Aron Virginas-Tar <aron.virginas-tar@arm.com> Mon Feb 18 14:49:04 2019 +0000
tree c8eaeef557d1a8a0e5a18a7104f5b6d308c9efc2
parent 2ee88dfe7096f8f571ae7be9cbf0f49ededd89af

IVGCVSW-2644 Add Serializer & Deserializer for Softmax

Change-Id: Ifea2108e173d2b602162fe53b880a68e1c715510
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>

src/armnnSerializer/Serializer.hpp

diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp
index 8a509e8..ec26dc1 100644
--- a/src/armnnSerializer/Serializer.hpp
+++ b/src/armnnSerializer/Serializer.hpp
@@ -9,7 +9,6 @@
 
 #include <armnnSerializer/ISerializer.hpp>
 
-#include <iostream>
 #include <unordered_map>
 
 #include <Schema_generated.h>
@@ -57,6 +56,10 @@
     void VisitMultiplicationLayer(const armnn::IConnectableLayer* layer,
                                   const char* name = nullptr) override;
 
+    void VisitSoftmaxLayer(const armnn::IConnectableLayer* layer,
+                           const armnn::SoftmaxDescriptor& softmaxDescriptor,
+                           const char* name = nullptr) override;
+
 private:
 
     /// Creates the Input Slots and Output Slots and LayerBase for the layer.