12 #include <arm_compute/runtime/CL/functions/CLLSTMLayerQuantized.h> 18 const TensorInfo& previousOutputIn,
const TensorInfo& cellStateOut,
19 const TensorInfo& output,
20 const QuantizedLstmInputParamsInfo& paramsInfo);
29 mutable arm_compute::CLLSTMLayerQuantized m_QuantizedLstmLayer;
31 std::unique_ptr<arm_compute::CLTensor> m_InputToInputWeightsTensor;
32 std::unique_ptr<arm_compute::CLTensor> m_InputToForgetWeightsTensor;
33 std::unique_ptr<arm_compute::CLTensor> m_InputToCellWeightsTensor;
34 std::unique_ptr<arm_compute::CLTensor> m_InputToOutputWeightsTensor;
35 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToInputWeightsTensor;
36 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToForgetWeightsTensor;
37 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToCellWeightsTensor;
38 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToOutputWeightsTensor;
39 std::unique_ptr<arm_compute::CLTensor> m_InputGateBiasTensor;
40 std::unique_ptr<arm_compute::CLTensor> m_ForgetGateBiasTensor;
41 std::unique_ptr<arm_compute::CLTensor> m_CellBiasTensor;
42 std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor;
44 void FreeUnusedTensors();
arm_compute::Status ClQuantizedLstmWorkloadValidate(const TensorInfo &input, const TensorInfo &previousCellStateIn, const TensorInfo &previousOutputIn, const TensorInfo &cellStateOut, const TensorInfo &output, const QuantizedLstmInputParamsInfo ¶msInfo)
void Execute() const override
ClQuantizedLstmWorkload(const QuantizedLstmQueueDescriptor &descriptor, const WorkloadInfo &info)