13 #include <arm_compute/runtime/CL/functions/CLLSTMLayer.h> 23 const arm_compute::CLCompileContext& clCompileContext);
27 mutable arm_compute::CLLSTMLayer m_LstmLayer;
29 std::unique_ptr<arm_compute::CLTensor> m_InputToInputWeightsTensor;
30 std::unique_ptr<arm_compute::CLTensor> m_InputToForgetWeightsTensor;
31 std::unique_ptr<arm_compute::CLTensor> m_InputToCellWeightsTensor;
32 std::unique_ptr<arm_compute::CLTensor> m_InputToOutputWeightsTensor;
33 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToInputWeightsTensor;
34 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToForgetWeightsTensor;
35 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToCellWeightsTensor;
36 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToOutputWeightsTensor;
37 std::unique_ptr<arm_compute::CLTensor> m_CellToInputWeightsTensor;
38 std::unique_ptr<arm_compute::CLTensor> m_CellToForgetWeightsTensor;
39 std::unique_ptr<arm_compute::CLTensor> m_CellToOutputWeightsTensor;
40 std::unique_ptr<arm_compute::CLTensor> m_InputGateBiasTensor;
41 std::unique_ptr<arm_compute::CLTensor> m_ForgetGateBiasTensor;
42 std::unique_ptr<arm_compute::CLTensor> m_CellBiasTensor;
43 std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor;
44 std::unique_ptr<arm_compute::CLTensor> m_ProjectionWeightsTensor;
45 std::unique_ptr<arm_compute::CLTensor> m_ProjectionBiasTensor;
46 std::unique_ptr<arm_compute::CLTensor> m_InputLayerNormWeightsTensor;
47 std::unique_ptr<arm_compute::CLTensor> m_ForgetLayerNormWeightsTensor;
48 std::unique_ptr<arm_compute::CLTensor> m_CellLayerNormWeightsTensor;
49 std::unique_ptr<arm_compute::CLTensor> m_OutputLayerNormWeightsTensor;
51 std::unique_ptr<arm_compute::CLTensor> m_ScratchBuffer;
53 void FreeUnusedTensors();
ClLstmFloatWorkload(const LstmQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
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)
Copyright (c) 2021 ARM Limited and Contributors.
An LstmDescriptor for the LstmLayer.
Contains information about inputs and outputs to a layer.
void Execute() const override