From 11af375a5a6bf88b4f3b933a86d53000b0d91ed0 Mon Sep 17 00:00:00 2001 From: Jim Flynn Date: Tue, 19 Mar 2019 17:22:29 +0000 Subject: IVGCVSW-2694: serialize/deserialize LSTM * added serialize/deserialize methods for LSTM and tests Change-Id: Ic59557f03001c496008c4bef92c2e0406e1fbc6c Signed-off-by: Nina Drozd Signed-off-by: Jim Flynn --- src/armnnSerializer/Serializer.hpp | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'src/armnnSerializer/Serializer.hpp') diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp index 71066d2699..4573bfdd11 100644 --- a/src/armnnSerializer/Serializer.hpp +++ b/src/armnnSerializer/Serializer.hpp @@ -111,6 +111,11 @@ public: const armnn::L2NormalizationDescriptor& l2NormalizationDescriptor, const char* name = nullptr) override; + void VisitLstmLayer(const armnn::IConnectableLayer* layer, + const armnn::LstmDescriptor& descriptor, + const armnn::LstmInputParams& params, + const char* name = nullptr) override; + void VisitMeanLayer(const armnn::IConnectableLayer* layer, const armnn::MeanDescriptor& descriptor, const char* name) override; -- cgit v1.2.1