From a2ec9092f0bff018bfe7ae0cacb7e30bcc17c1c7 Mon Sep 17 00:00:00 2001
From: Jan Eilers <jan.eilers@arm.com>
Date: Mon, 8 Jul 2019 15:56:59 +0100
Subject: 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 | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'src/backends/cl/workloads/ClLstmFloatWorkload.hpp')

diff --git a/src/backends/cl/workloads/ClLstmFloatWorkload.hpp b/src/backends/cl/workloads/ClLstmFloatWorkload.hpp
index 9a3211a037..5bd67c256f 100644
--- a/src/backends/cl/workloads/ClLstmFloatWorkload.hpp
+++ b/src/backends/cl/workloads/ClLstmFloatWorkload.hpp
@@ -39,6 +39,10 @@ private:
     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;
 
-- 
cgit v1.2.1