19 : BaseWorkload<QuantizedLstmQueueDescriptor>(descriptor,
info)
22 m_InputToInputWeightsTensor = std::make_unique<arm_compute::Tensor>();
25 m_InputToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>();
28 m_InputToCellWeightsTensor = std::make_unique<arm_compute::Tensor>();
31 m_InputToOutputWeightsTensor = std::make_unique<arm_compute::Tensor>();
34 m_RecurrentToInputWeightsTensor = std::make_unique<arm_compute::Tensor>();
37 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>();
40 m_RecurrentToCellWeightsTensor = std::make_unique<arm_compute::Tensor>();
43 m_RecurrentToOutputWeightsTensor = std::make_unique<arm_compute::Tensor>();
46 m_InputGateBiasTensor = std::make_unique<arm_compute::Tensor>();
49 m_ForgetGateBiasTensor = std::make_unique<arm_compute::Tensor>();
52 m_CellBiasTensor = std::make_unique<arm_compute::Tensor>();
55 m_OutputGateBiasTensor = std::make_unique<arm_compute::Tensor>();
58 const arm_compute::ITensor& input =
static_cast<IAclTensorHandle*
>(
m_Data.
m_Inputs[0])->GetTensor();
59 arm_compute::ITensor& cell_state_in =
static_cast<IAclTensorHandle*
>(
m_Data.
m_Inputs[1])->GetTensor();
60 const arm_compute::ITensor& output_state_in =
static_cast<IAclTensorHandle*
>(
m_Data.
m_Inputs[2])->GetTensor();
62 arm_compute::ITensor& cell_state_out =
static_cast<IAclTensorHandle*
>(
m_Data.
m_Outputs[0])->GetTensor();
63 arm_compute::ITensor& output_state_out =
static_cast<IAclTensorHandle*
>(
m_Data.
m_Outputs[1])->GetTensor();
65 m_QuantizedLstmLayer.configure(&input,
66 m_InputToInputWeightsTensor.get(),
67 m_InputToForgetWeightsTensor.get(),
68 m_InputToCellWeightsTensor.get(),
69 m_InputToOutputWeightsTensor.get(),
70 m_RecurrentToInputWeightsTensor.get(),
71 m_RecurrentToForgetWeightsTensor.get(),
72 m_RecurrentToCellWeightsTensor.get(),
73 m_RecurrentToOutputWeightsTensor.get(),
74 m_InputGateBiasTensor.get(),
75 m_ForgetGateBiasTensor.get(),
76 m_CellBiasTensor.get(),
77 m_OutputGateBiasTensor.get(),
121 m_QuantizedLstmLayer.prepare();
const ConstCpuTensorHandle * m_CellBias
const ConstCpuTensorHandle * m_ForgetGateBias
const ConstCpuTensorHandle * m_InputToForgetWeights
const TensorInfo & GetTensorInfo() const
const ConstCpuTensorHandle * m_InputToInputWeights
const ConstCpuTensorHandle * m_RecurrentToInputWeights
const ConstCpuTensorHandle * m_OutputGateBias
const ConstCpuTensorHandle * m_InputGateBias
const QuantizedLstmQueueDescriptor m_Data
const ConstCpuTensorHandle * m_RecurrentToForgetWeights
const ConstCpuTensorHandle * m_RecurrentToOutputWeights
const ConstCpuTensorHandle * m_InputToOutputWeights
std::vector< ITensorHandle * > m_Outputs
std::vector< ITensorHandle * > m_Inputs
const ConstCpuTensorHandle * m_RecurrentToCellWeights
void InitializeArmComputeTensorData(arm_compute::Tensor &tensor, const ConstCpuTensorHandle *handle)
const ConstCpuTensorHandle * m_InputToCellWeights