diff options
Diffstat (limited to 'src/backends/reference/workloads/RefLstmWorkload.hpp')
-rw-r--r-- | src/backends/reference/workloads/RefLstmWorkload.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/RefLstmWorkload.hpp b/src/backends/reference/workloads/RefLstmWorkload.hpp index 38e3fb956c..ce5a775269 100644 --- a/src/backends/reference/workloads/RefLstmWorkload.hpp +++ b/src/backends/reference/workloads/RefLstmWorkload.hpp @@ -38,6 +38,12 @@ private: std::unique_ptr<ScopedCpuTensorHandle> m_OutputGateBiasTensor; std::unique_ptr<ScopedCpuTensorHandle> m_ProjectionWeightsTensor; std::unique_ptr<ScopedCpuTensorHandle> m_ProjectionBiasTensor; + std::unique_ptr<ScopedCpuTensorHandle> m_InputLayerNormWeights; + std::unique_ptr<ScopedCpuTensorHandle> m_ForgetLayerNormWeights; + std::unique_ptr<ScopedCpuTensorHandle> m_CellLayerNormWeights; + std::unique_ptr<ScopedCpuTensorHandle> m_OutputLayerNormWeights; + + float m_LayerNormEpsilon = static_cast<float>(1e-8); }; } //namespace armnn |