diff options
Diffstat (limited to 'src/armnn/layers/LstmLayer.hpp')
-rw-r--r-- | src/armnn/layers/LstmLayer.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnn/layers/LstmLayer.hpp b/src/armnn/layers/LstmLayer.hpp index 21421f220f..5ccb4bcf92 100644 --- a/src/armnn/layers/LstmLayer.hpp +++ b/src/armnn/layers/LstmLayer.hpp @@ -30,8 +30,6 @@ struct LstmOptCifgParameters /// A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units]. std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToInputWeights; /// A unique pointer to represent 1D weights tensor with dimensions [num_units]. - std::unique_ptr<ScopedCpuTensorHandle> m_CellToInputWeights; - /// A unique pointer to represent 1D weights tensor with dimensions [num_units]. std::unique_ptr<ScopedCpuTensorHandle> m_InputGateBias; }; @@ -46,6 +44,8 @@ struct LstmOptProjectionParameters struct LstmOptPeepholeParameters { /// A unique pointer to represent 1D weights tensor with dimensions [num_units]. + std::unique_ptr<ScopedCpuTensorHandle> m_CellToInputWeights; + /// A unique pointer to represent 1D weights tensor with dimensions [num_units]. std::unique_ptr<ScopedCpuTensorHandle> m_CellToForgetWeights; /// A unique pointer to represent 1D weights tensor with dimensions [num_units]. std::unique_ptr<ScopedCpuTensorHandle> m_CellToOutputWeights; |