13 #include "arm_compute/graph/Tensor.h" 14 #include "arm_compute/runtime/NEON/functions/NELSTMLayer.h" 23 virtual void Execute()
const override;
26 mutable arm_compute::NELSTMLayer m_LstmLayer;
28 std::unique_ptr<arm_compute::Tensor> m_InputToInputWeightsTensor;
29 std::unique_ptr<arm_compute::Tensor> m_InputToForgetWeightsTensor;
30 std::unique_ptr<arm_compute::Tensor> m_InputToCellWeightsTensor;
31 std::unique_ptr<arm_compute::Tensor> m_InputToOutputWeightsTensor;
32 std::unique_ptr<arm_compute::Tensor> m_RecurrentToInputWeightsTensor;
33 std::unique_ptr<arm_compute::Tensor> m_RecurrentToForgetWeightsTensor;
34 std::unique_ptr<arm_compute::Tensor> m_RecurrentToCellWeightsTensor;
35 std::unique_ptr<arm_compute::Tensor> m_RecurrentToOutputWeightsTensor;
36 std::unique_ptr<arm_compute::Tensor> m_CellToInputWeightsTensor;
37 std::unique_ptr<arm_compute::Tensor> m_CellToForgetWeightsTensor;
38 std::unique_ptr<arm_compute::Tensor> m_CellToOutputWeightsTensor;
39 std::unique_ptr<arm_compute::Tensor> m_InputGateBiasTensor;
40 std::unique_ptr<arm_compute::Tensor> m_ForgetGateBiasTensor;
41 std::unique_ptr<arm_compute::Tensor> m_CellBiasTensor;
42 std::unique_ptr<arm_compute::Tensor> m_OutputGateBiasTensor;
43 std::unique_ptr<arm_compute::Tensor> m_ProjectionWeightsTensor;
44 std::unique_ptr<arm_compute::Tensor> m_ProjectionBiasTensor;
46 std::unique_ptr<arm_compute::Tensor> m_ScratchBuffer;
48 std::unique_ptr<arm_compute::Tensor> m_InputLayerNormWeightsTensor;
49 std::unique_ptr<arm_compute::Tensor> m_ForgetLayerNormWeightsTensor;
50 std::unique_ptr<arm_compute::Tensor> m_CellLayerNormWeightsTensor;
51 std::unique_ptr<arm_compute::Tensor> m_OutputLayerNormWeightsTensor;
53 void FreeUnusedTensors();
virtual void Execute() const override
An LstmDescriptor for the LstmLayer.
arm_compute::Status NeonLstmFloatWorkloadValidate(const TensorInfo &input, const TensorInfo &outputStateIn, const TensorInfo &cellStateIn, const TensorInfo &scratchBuffer, const TensorInfo &outputStateOut, const TensorInfo &cellStateOut, const TensorInfo &output, const LstmDescriptor &descriptor, const LstmInputParamsInfo ¶msInfo)
NeonLstmFloatWorkload(const LstmQueueDescriptor &descriptor, const WorkloadInfo &info)