diff options
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. |