From ee18dc8d1725f472850ab0c398fd7cbc4b850891 Mon Sep 17 00:00:00 2001 From: James Conroy Date: Wed, 17 Jul 2019 11:27:46 +0100 Subject: IVGCVSW-3469 Add front end for Quantized LSTM layer * Added new layer QuantizedLstm (Android Q) * Made necessary changes to APIs * Added unit tests Change-Id: I3b9f16b0e7e49f51932cf204c87cb7118798123a Signed-off-by: James Conroy --- src/armnnSerializer/Serializer.hpp | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'src/armnnSerializer/Serializer.hpp') diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp index 8404a7f093..b859ae9cf0 100644 --- a/src/armnnSerializer/Serializer.hpp +++ b/src/armnnSerializer/Serializer.hpp @@ -166,6 +166,10 @@ public: void VisitQuantizeLayer(const armnn::IConnectableLayer* layer, const char* name = nullptr) override; + void VisitQuantizedLstmLayer(const armnn::IConnectableLayer* layer, + const armnn::QuantizedLstmInputParams& params, + const char* name = nullptr) override; + void VisitReshapeLayer(const armnn::IConnectableLayer* layer, const armnn::ReshapeDescriptor& reshapeDescriptor, const char* name = nullptr) override; -- cgit v1.2.1