67 BaseWorkload<QuantizedLstmQueueDescriptor>(descriptor,
info)
69 m_InputToInputWeightsTensor = std::make_unique<arm_compute::CLTensor>();
72 m_InputToForgetWeightsTensor = std::make_unique<arm_compute::CLTensor>();
75 m_InputToCellWeightsTensor = std::make_unique<arm_compute::CLTensor>();
78 m_InputToOutputWeightsTensor = std::make_unique<arm_compute::CLTensor>();
81 m_RecurrentToInputWeightsTensor = std::make_unique<arm_compute::CLTensor>();
84 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::CLTensor>();
87 m_RecurrentToCellWeightsTensor = std::make_unique<arm_compute::CLTensor>();
90 m_RecurrentToOutputWeightsTensor = std::make_unique<arm_compute::CLTensor>();
93 m_InputGateBiasTensor = std::make_unique<arm_compute::CLTensor>();
96 m_ForgetGateBiasTensor = std::make_unique<arm_compute::CLTensor>();
99 m_CellBiasTensor = std::make_unique<arm_compute::CLTensor>();
102 m_OutputGateBiasTensor = std::make_unique<arm_compute::CLTensor>();
105 const arm_compute::ICLTensor& inputTensor =
static_cast<IClTensorHandle*
>(
m_Data.
m_Inputs[0])->GetTensor();
106 arm_compute::ICLTensor& cellStateInTensor =
static_cast<IClTensorHandle*
>(
m_Data.
m_Inputs[1])->GetTensor();
107 const arm_compute::ICLTensor& outputStateInTensor =
static_cast<IClTensorHandle*
>(
m_Data.
m_Inputs[2])->GetTensor();
109 arm_compute::ICLTensor& cellStateOutTensor =
static_cast<IClTensorHandle*
>(
m_Data.
m_Outputs[0])->GetTensor();
110 arm_compute::ICLTensor& outputStateOutTensor =
static_cast<IClTensorHandle*
>(
m_Data.
m_Outputs[1])->GetTensor();
112 m_QuantizedLstmLayer.configure(clCompileContext, &inputTensor, m_InputToInputWeightsTensor.get(),
113 m_InputToForgetWeightsTensor.get(),
114 m_InputToCellWeightsTensor.get(), m_InputToOutputWeightsTensor.get(),
115 m_RecurrentToInputWeightsTensor.get(), m_RecurrentToForgetWeightsTensor.get(),
116 m_RecurrentToCellWeightsTensor.get(), m_RecurrentToOutputWeightsTensor.get(),
117 m_InputGateBiasTensor.get(), m_ForgetGateBiasTensor.get(), m_CellBiasTensor.get(),
118 m_OutputGateBiasTensor.get(), &cellStateInTensor, &outputStateInTensor,
119 &cellStateOutTensor, &outputStateOutTensor);
134 m_QuantizedLstmLayer.prepare();
const ConstTensorHandle * m_InputGateBias
const ConstTensorHandle * m_RecurrentToInputWeights
const TensorInfo & GetTensorInfo() const
const ConstTensorHandle * m_InputToForgetWeights
QuantizedLstmQueueDescriptor m_Data
const ConstTensorHandle * m_RecurrentToCellWeights
const ConstTensorHandle * m_ForgetGateBias
const ConstTensorHandle * m_RecurrentToOutputWeights
const ConstTensorHandle * m_OutputGateBias
const ConstTensorHandle * m_RecurrentToForgetWeights
const ConstTensorHandle * m_InputToOutputWeights
std::vector< ITensorHandle * > m_Outputs
const ConstTensorHandle * m_InputToInputWeights
const ConstTensorHandle * m_CellBias
void InitializeArmComputeClTensorData(arm_compute::CLTensor &clTensor, const ConstTensorHandle *handle)
std::vector< ITensorHandle * > m_Inputs
const ConstTensorHandle * m_InputToCellWeights