From a0162e17c56538ee6d72ecce4c3e0836cbb34c56 Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Fri, 23 Jul 2021 14:47:49 +0100 Subject: MLCE-530 Add Serializer and Deserializer for UnidirectionalSequenceLstm Signed-off-by: Narumol Prangnawarat Change-Id: Ic1c56a57941ebede19ab8b9032e7f9df1221be7a --- 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 c99e87d3e9..dead8739cc 100644 --- a/src/armnnSerializer/Serializer.hpp +++ b/src/armnnSerializer/Serializer.hpp @@ -348,6 +348,11 @@ private: void SerializeTransposeLayer(const armnn::IConnectableLayer* layer, const armnn::TransposeDescriptor& descriptor, const char* name = nullptr); + + void SerializeUnidirectionalSequenceLstmLayer(const armnn::IConnectableLayer* layer, + const armnn::UnidirectionalSequenceLstmDescriptor& descriptor, + const std::vector& constants, + const char* name = nullptr); }; -- cgit v1.2.1