commit a2ec9092f0bff018bfe7ae0cacb7e30bcc17c1c7
author Jan Eilers <jan.eilers@arm.com> Mon Jul 08 15:56:59 2019 +0100
committer Nikhil Raj Arm <nikhil.raj@arm.com> Tue Jul 09 15:08:10 2019 +0000
tree 11a36098492aed1629d873dcc02fb25a0071de2b
parent c0ed7baa8c05c4710034dfd179fadd31b716a46f

IVGCVSW-3338 Add CL backend support for LSTM normalization

 * Enable calls to LSTM normalization unit tests on CL backend.
 * Update CL workload to set the layer normalization parameters.

!android-nn-driver:1461

Change-Id: Ia5a29918961c391c1f1d8f331add377a38822ddd
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Signed-off-by: Jan Eilers <jan.eilers@arm.com>

src/backends/cl/workloads/ClLstmFloatWorkload.hpp

diff --git a/src/backends/cl/workloads/ClLstmFloatWorkload.hpp b/src/backends/cl/workloads/ClLstmFloatWorkload.hpp
index 9a3211a..5bd67c2 100644
--- a/src/backends/cl/workloads/ClLstmFloatWorkload.hpp
+++ b/src/backends/cl/workloads/ClLstmFloatWorkload.hpp
@@ -39,6 +39,10 @@
     std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor;
     std::unique_ptr<arm_compute::CLTensor> m_ProjectionWeightsTensor;
     std::unique_ptr<arm_compute::CLTensor> m_ProjectionBiasTensor;
+    std::unique_ptr<arm_compute::CLTensor> m_InputLayerNormWeightsTensor;
+    std::unique_ptr<arm_compute::CLTensor> m_ForgetLayerNormWeightsTensor;
+    std::unique_ptr<arm_compute::CLTensor> m_CellLayerNormWeightsTensor;
+    std::unique_ptr<arm_compute::CLTensor> m_OutputLayerNormWeightsTensor;
 
     std::unique_ptr<arm_compute::CLTensor> m_ScratchBuffer;