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;
29 std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToInputWeightsTensor;
30 std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToForgetWeightsTensor;
31 std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToCellWeightsTensor;
32 std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToOutputWeightsTensor;
34 std::unique_ptr<ScopedCpuTensorHandle> m_CellToInputWeightsTensor;
35 std::unique_ptr<ScopedCpuTensorHandle> m_CellToForgetWeightsTensor;
36 std::unique_ptr<ScopedCpuTensorHandle> m_CellToOutputWeightsTensor;
38 std::unique_ptr<ScopedCpuTensorHandle> m_InputGateBiasTensor;
39 std::unique_ptr<ScopedCpuTensorHandle> m_ForgetGateBiasTensor;
40 std::unique_ptr<ScopedCpuTensorHandle> m_CellBiasTensor;
41 std::unique_ptr<ScopedCpuTensorHandle> m_OutputGateBiasTensor;
43 std::unique_ptr<ScopedCpuTensorHandle> m_ProjectionWeightsTensor;
44 std::unique_ptr<ScopedCpuTensorHandle> m_ProjectionBiasTensor;
46 std::unique_ptr<ScopedCpuTensorHandle> m_InputLayerNormWeightsTensor;
47 std::unique_ptr<ScopedCpuTensorHandle> m_ForgetLayerNormWeightsTensor;
48 std::unique_ptr<ScopedCpuTensorHandle> m_CellLayerNormWeightsTensor;
49 std::unique_ptr<ScopedCpuTensorHandle> m_OutputLayerNormWeightsTensor;
51 float m_LayerNormEpsilon =
static_cast<float>(1e-8);
virtual void Execute() const override
Copyright (c) 2020 ARM Limited.
RefQLstmWorkload(const QLstmQueueDescriptor &descriptor, const WorkloadInfo &info)
Contains information about inputs and outputs to a layer.