12 #include <arm_compute/runtime/CL/functions/CLLSTMLayerQuantized.h> 18 const TensorInfo& previousOutputIn,
const TensorInfo& cellStateOut,
19 const TensorInfo& output,
20 const QuantizedLstmInputParamsInfo& paramsInfo);
27 const arm_compute::CLCompileContext& clCompileContext);
31 mutable arm_compute::CLLSTMLayerQuantized m_QuantizedLstmLayer;
33 std::unique_ptr<arm_compute::CLTensor> m_InputToInputWeightsTensor;
34 std::unique_ptr<arm_compute::CLTensor> m_InputToForgetWeightsTensor;
35 std::unique_ptr<arm_compute::CLTensor> m_InputToCellWeightsTensor;
36 std::unique_ptr<arm_compute::CLTensor> m_InputToOutputWeightsTensor;
37 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToInputWeightsTensor;
38 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToForgetWeightsTensor;
39 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToCellWeightsTensor;
40 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToOutputWeightsTensor;
41 std::unique_ptr<arm_compute::CLTensor> m_InputGateBiasTensor;
42 std::unique_ptr<arm_compute::CLTensor> m_ForgetGateBiasTensor;
43 std::unique_ptr<arm_compute::CLTensor> m_CellBiasTensor;
44 std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor;
46 void FreeUnusedTensors();
Copyright (c) 2021 ARM Limited and Contributors.
ClQuantizedLstmWorkload(const QuantizedLstmQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
void Execute() const override
Contains information about TensorInfos of a layer.
arm_compute::Status ClQuantizedLstmWorkloadValidate(const TensorInfo &input, const TensorInfo &previousCellStateIn, const TensorInfo &previousOutputIn, const TensorInfo &cellStateOut, const TensorInfo &output, const QuantizedLstmInputParamsInfo ¶msInfo)