IVGCVSW-6268 Add support of Unidirectional Sequence Lstm fp32/fp16 to Neon

!ComputeLibrary:7150

Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: I3de48ffc8d08c95a22705e2b68d069791bddae73

1 files changed, 10 insertions, 0 deletions

diff --git a/src/backends/neon/NeonLayerSupport.hpp b/src/backends/neon/NeonLayerSupport.hpp
index afa9b419e6..1eef41fda5 100644
--- a/src/backends/neon/NeonLayerSupport.hpp
+++ b/src/backends/neon/NeonLayerSupport.hpp
@@ -336,6 +336,16 @@ public:
                               const TransposeDescriptor& descriptor,
                               Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
 
+    bool IsUnidirectionalSequenceLstmSupported(const TensorInfo& input,
+                                               const TensorInfo& outputStateIn,
+                                               const TensorInfo& cellStateIn,
+                                               const TensorInfo& output,
+                                               const Optional<TensorInfo>& hiddenStateOutput,
+                                               const Optional<TensorInfo>& cellStateOutput,
+                                               const UnidirectionalSequenceLstmDescriptor& descriptor,
+                                               const LstmInputParamsInfo& paramsInfo,
+                                               Optional<std::string&> reasonIfUnsupported) const override;
+
 private:
     const IBackendInternal::IBackendSpecificModelContextPtr m_ModelContextPtr;