IVGCVSW-3236 Extend Ref LSTM with layer normalization support

* Add descriptor values
* Update lstm queue descriptor validate function
* Update lstm workload
* Update isLstmSupported (Cl and Ref), LayerSupportBase, ILayerSupport
* Update lstm layer
* Add unit tests

Signed-off-by: Jan Eilers <jan.eilers@arm.com>
Change-Id: I932175d550facfb342325051eaa7bd2084ebdc18
Signed-off-by: Jan Eilers <jan.eilers@arm.com>

1 files changed, 8 insertions, 0 deletions

diff --git a/include/armnn/LstmParams.hpp b/include/armnn/LstmParams.hpp
index c4f38f0067..a7c57c78b2 100644
--- a/include/armnn/LstmParams.hpp
+++ b/include/armnn/LstmParams.hpp
@@ -29,6 +29,10 @@ struct LstmInputParams
         , m_OutputGateBias(nullptr)
         , m_ProjectionWeights(nullptr)
         , m_ProjectionBias(nullptr)
+        , m_InputLayerNormWeights(nullptr)
+        , m_ForgetLayerNormWeights(nullptr)
+        , m_CellLayerNormWeights(nullptr)
+        , m_OutputLayerNormWeights(nullptr)
     {
     }
 
@@ -49,6 +53,10 @@ struct LstmInputParams
     const ConstTensor* m_OutputGateBias;
     const ConstTensor* m_ProjectionWeights;
     const ConstTensor* m_ProjectionBias;
+    const ConstTensor* m_InputLayerNormWeights;
+    const ConstTensor* m_ForgetLayerNormWeights;
+    const ConstTensor* m_CellLayerNormWeights;
+    const ConstTensor* m_OutputLayerNormWeights;
 };
 
 } // namespace armnn