21 virtual void Execute()
const override;
24 std::unique_ptr<ScopedCpuTensorHandle> m_InputToInputWeightsTensor;
25 std::unique_ptr<ScopedCpuTensorHandle> m_InputToForgetWeightsTensor;
26 std::unique_ptr<ScopedCpuTensorHandle> m_InputToCellWeightsTensor;
27 std::unique_ptr<ScopedCpuTensorHandle> m_InputToOutputWeightsTensor;
28 std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToInputWeightsTensor;
29 std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToForgetWeightsTensor;
30 std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToCellWeightsTensor;
31 std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToOutputWeightsTensor;
32 std::unique_ptr<ScopedCpuTensorHandle> m_CellToInputWeightsTensor;
33 std::unique_ptr<ScopedCpuTensorHandle> m_CellToForgetWeightsTensor;
34 std::unique_ptr<ScopedCpuTensorHandle> m_CellToOutputWeightsTensor;
35 std::unique_ptr<ScopedCpuTensorHandle> m_InputGateBiasTensor;
36 std::unique_ptr<ScopedCpuTensorHandle> m_ForgetGateBiasTensor;
37 std::unique_ptr<ScopedCpuTensorHandle> m_CellBiasTensor;
38 std::unique_ptr<ScopedCpuTensorHandle> m_OutputGateBiasTensor;
39 std::unique_ptr<ScopedCpuTensorHandle> m_ProjectionWeightsTensor;
40 std::unique_ptr<ScopedCpuTensorHandle> m_ProjectionBiasTensor;
41 std::unique_ptr<ScopedCpuTensorHandle> m_InputLayerNormWeights;
42 std::unique_ptr<ScopedCpuTensorHandle> m_ForgetLayerNormWeights;
43 std::unique_ptr<ScopedCpuTensorHandle> m_CellLayerNormWeights;
44 std::unique_ptr<ScopedCpuTensorHandle> m_OutputLayerNormWeights;
46 float m_LayerNormEpsilon =
static_cast<float>(1e-8);
RefLstmWorkload(const LstmQueueDescriptor &descriptor, const WorkloadInfo &info)
Copyright (c) 2020 ARM Limited.
virtual void Execute() const override
Contains information about inputs and outputs to a layer.