diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-02-13 15:41:52 +0000 |
---|---|---|
committer | Aron Virginas-Tar <aron.virginas-tar@arm.com> | 2019-02-18 14:49:04 +0000 |
commit | fc413c0c977e6c9680a2aa6546e977be0a2efdb9 (patch) | |
tree | c8eaeef557d1a8a0e5a18a7104f5b6d308c9efc2 /src/armnnSerializer/Serializer.hpp | |
parent | 2ee88dfe7096f8f571ae7be9cbf0f49ededd89af (diff) | |
download | armnn-fc413c0c977e6c9680a2aa6546e977be0a2efdb9.tar.gz |
IVGCVSW-2644 Add Serializer & Deserializer for Softmax
Change-Id: Ifea2108e173d2b602162fe53b880a68e1c715510
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Diffstat (limited to 'src/armnnSerializer/Serializer.hpp')
-rw-r--r-- | src/armnnSerializer/Serializer.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp index 8a509e880e..ec26dc1796 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 @@ public: 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. |