diff options
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; |