diff options
author | Mike Kelly <mike.kelly@arm.com> | 2019-02-20 16:53:11 +0000 |
---|---|---|
committer | Mike Kelly <mike.kelly@arm.com> | 2019-02-20 16:53:11 +0000 |
commit | af484013329a8ca5b3c4c9d16395fb79dd19b1b2 (patch) | |
tree | 90a6e08d99a3856403c79395cd4b58bad8755e4a /src/armnnSerializer/Serializer.hpp | |
parent | 0028d1b0ce5f4c2c6a6eb3c66f38111c21eb47a3 (diff) | |
download | armnn-af484013329a8ca5b3c4c9d16395fb79dd19b1b2.tar.gz |
IVGCVSW-2643 Add Serializer & Deserializer for Activation
* Added ActivationLayer to Schema.fbs
* Added Activation serialization and deserialization support
* Added serialization and deserialization unit tests
Change-Id: Ib5df45f123674988b994ffe3f111d3fb57864912
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Diffstat (limited to 'src/armnnSerializer/Serializer.hpp')
-rw-r--r-- | src/armnnSerializer/Serializer.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp index aa765a2065..0c442e0788 100644 --- a/src/armnnSerializer/Serializer.hpp +++ b/src/armnnSerializer/Serializer.hpp @@ -42,6 +42,10 @@ public: return m_serializedLayers; } + void VisitActivationLayer(const armnn::IConnectableLayer* layer, + const armnn::ActivationDescriptor& descriptor, + const char* name = nullptr) override; + void VisitAdditionLayer(const armnn::IConnectableLayer* layer, const char* name = nullptr) override; |