diff options
author | Jan Eilers <jan.eilers@arm.com> | 2019-07-08 15:56:59 +0100 |
---|---|---|
committer | Nikhil Raj Arm <nikhil.raj@arm.com> | 2019-07-09 15:08:10 +0000 |
commit | a2ec9092f0bff018bfe7ae0cacb7e30bcc17c1c7 (patch) | |
tree | 11a36098492aed1629d873dcc02fb25a0071de2b /src/backends/cl/workloads/ClLstmFloatWorkload.hpp | |
parent | c0ed7baa8c05c4710034dfd179fadd31b716a46f (diff) | |
download | armnn-a2ec9092f0bff018bfe7ae0cacb7e30bcc17c1c7.tar.gz |
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>
Diffstat (limited to 'src/backends/cl/workloads/ClLstmFloatWorkload.hpp')
-rw-r--r-- | src/backends/cl/workloads/ClLstmFloatWorkload.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
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; |