13 #include <arm_compute/runtime/CL/functions/CLLSTMLayer.h> 25 mutable arm_compute::CLLSTMLayer m_LstmLayer;
27 std::unique_ptr<arm_compute::CLTensor> m_InputToInputWeightsTensor;
28 std::unique_ptr<arm_compute::CLTensor> m_InputToForgetWeightsTensor;
29 std::unique_ptr<arm_compute::CLTensor> m_InputToCellWeightsTensor;
30 std::unique_ptr<arm_compute::CLTensor> m_InputToOutputWeightsTensor;
31 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToInputWeightsTensor;
32 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToForgetWeightsTensor;
33 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToCellWeightsTensor;
34 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToOutputWeightsTensor;
35 std::unique_ptr<arm_compute::CLTensor> m_CellToInputWeightsTensor;
36 std::unique_ptr<arm_compute::CLTensor> m_CellToForgetWeightsTensor;
37 std::unique_ptr<arm_compute::CLTensor> m_CellToOutputWeightsTensor;
38 std::unique_ptr<arm_compute::CLTensor> m_InputGateBiasTensor;
39 std::unique_ptr<arm_compute::CLTensor> m_ForgetGateBiasTensor;
40 std::unique_ptr<arm_compute::CLTensor> m_CellBiasTensor;
41 std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor;
42 std::unique_ptr<arm_compute::CLTensor> m_ProjectionWeightsTensor;
43 std::unique_ptr<arm_compute::CLTensor> m_ProjectionBiasTensor;
44 std::unique_ptr<arm_compute::CLTensor> m_InputLayerNormWeightsTensor;
45 std::unique_ptr<arm_compute::CLTensor> m_ForgetLayerNormWeightsTensor;
46 std::unique_ptr<arm_compute::CLTensor> m_CellLayerNormWeightsTensor;
47 std::unique_ptr<arm_compute::CLTensor> m_OutputLayerNormWeightsTensor;
49 std::unique_ptr<arm_compute::CLTensor> m_ScratchBuffer;
51 void FreeUnusedTensors();
An LstmDescriptor for the LstmLayer.
void Execute() const override
arm_compute::Status ClLstmFloatWorkloadValidate(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)
ClLstmFloatWorkload(const LstmQueueDescriptor &descriptor, const WorkloadInfo &info)