201     const int num_units   = input_to_output_weights->info()->dimension(1);  in configure()  local
253 …s->info(), _input_to_input_eff_bias.info(), GEMMLowpReductionKernelInfo(num_units, false, -qinput.…  in configure()
254 …weights->info(), _recurrent_to_input_eff_bias.info(), GEMMLowpReductionKernelInfo(num_units, false,  in configure()
257 …->info(), _input_to_forget_eff_bias.info(), GEMMLowpReductionKernelInfo(num_units, false, -qinput.…  in configure()
258 …eights->info(), _recurrent_to_forget_eff_bias.info(), GEMMLowpReductionKernelInfo(num_units, false,  in configure()
260 …ts->info(), _input_to_cell_eff_bias.info(), GEMMLowpReductionKernelInfo(num_units, false, -qinput.…  in configure()
261 …info(), _recurrent_to_cell_eff_bias.info(), GEMMLowpReductionKernelInfo(num_units, false, -qoutput…  in configure()
263 …->info(), _input_to_output_eff_bias.info(), GEMMLowpReductionKernelInfo(num_units, false, -qinput.…  in configure()
264 …eights->info(), _recurrent_to_output_eff_bias.info(), GEMMLowpReductionKernelInfo(num_units, false,  in configure()
298     const TensorInfo mm_out_info(TensorShape(num_units, batch_size), 1, DataType::S32);  in configure()
345 …const TensorInfo forget_gate_info(TensorShape(num_units, batch_size), 1, DataType::QSYMM16, sigmoi…  in configure()
378 …const TensorInfo cell_gate_info(TensorShape(num_units, batch_size), 1, DataType::QSYMM16, sigmoid_…  in configure()
385 …const TensorInfo input_gate_info(TensorShape(num_units, batch_size), 1, DataType::QSYMM16, sigmoid…  in configure()
396 …const TensorInfo input_outstage_info(TensorShape(num_units, batch_size), 1, DataType::QSYMM16, Qua…  in configure()
444 …const TensorInfo mul_input_cell_info(TensorShape(num_units, batch_size), 1, DataType::QSYMM16, Qua…  in configure()
457 …const TensorInfo output_outstage_info(TensorShape(num_units, batch_size), 1, DataType::QSYMM16, Qu…  in configure()
503 …const TensorInfo output_gate_info(TensorShape(num_units, batch_size), 1, DataType::QSYMM16, sigmoi…  in configure()
523     _projection_tensor_copy_required = (num_units != output_size);  in configure()
620     const unsigned int num_units   = input_to_output_weights->dimension(1);  in validate()  local
627     ARM_COMPUTE_RETURN_ERROR_ON(recurrent_to_output_weights->dimension(1) != num_units);  in validate()
634     ARM_COMPUTE_RETURN_ERROR_ON(forget_gate_bias->dimension(0) != num_units);  in validate()
640     ARM_COMPUTE_RETURN_ERROR_ON(cell_state_in->dimension(0) != num_units);  in validate()
655 …      ARM_COMPUTE_RETURN_ERROR_ON(lstm_params.cell_to_forget_weights()->dimension(0) != num_units);  in validate()
683     const TensorInfo eff_bias_info(TensorShape(num_units), 1, DataType::S32);  in validate()
687 …s.input_to_input_weights(), &eff_bias_info, GEMMLowpReductionKernelInfo(num_units, false, -qinput.…  in validate()
688 …current_to_input_weights(), &eff_bias_info, GEMMLowpReductionKernelInfo(num_units, false, -qoutput…  in validate()
691 …te(input_to_forget_weights, &eff_bias_info, GEMMLowpReductionKernelInfo(num_units, false, -qinput.…  in validate()
692 …ecurrent_to_forget_weights, &eff_bias_info, GEMMLowpReductionKernelInfo(num_units, false, -qoutput…  in validate()
693 …date(input_to_cell_weights, &eff_bias_info, GEMMLowpReductionKernelInfo(num_units, false, -qinput.…  in validate()
694 …(recurrent_to_cell_weights, &eff_bias_info, GEMMLowpReductionKernelInfo(num_units, false, -qoutput…  in validate()
695 …te(input_to_output_weights, &eff_bias_info, GEMMLowpReductionKernelInfo(num_units, false, -qinput.…  in validate()
696 …ecurrent_to_output_weights, &eff_bias_info, GEMMLowpReductionKernelInfo(num_units, false, -qoutput…  in validate()
710 …const TensorInfo input_weights_transposed(TensorShape(num_units, input_size), 1, input_to_forget_w…  in validate()
711 …const TensorInfo recurrent_weights_transposed(TensorShape(num_units, output_size), 1, recurrent_to…  in validate()
727 …const TensorInfo projection_weights_transposed(TensorShape(output_size, num_units), 1, lstm_params…  in validate()
741 …const TensorInfo forget_outstage_info(TensorShape(num_units, batch_size), 1, DataType::QSYMM16, Qu…  in validate()
742     const TensorInfo mm_out_info(TensorShape(num_units, batch_size), 1, DataType::S32);  in validate()
772 …const TensorInfo forget_gate_info(TensorShape(num_units, batch_size), 1, DataType::QSYMM16, sigmoi…  in validate()
777 …const TensorInfo cell_outstage_info(TensorShape(num_units, batch_size), 1, DataType::QSYMM16, Quan…  in validate()
793 …const TensorInfo cell_gate_info(TensorShape(num_units, batch_size), 1, DataType::QSYMM16, sigmoid_…  in validate()
797 …const TensorInfo input_gate_info(TensorShape(num_units, batch_size), 1, DataType::QSYMM16, sigmoid…  in validate()
813 …const TensorInfo input_outstage_info(TensorShape(num_units, batch_size), 1, DataType::QSYMM16, Qua…  in validate()
852 …const TensorInfo output_outstage_info(TensorShape(num_units, batch_size), 1, DataType::QSYMM16, Qu…  in validate()
879 …const TensorInfo output_gate_info(TensorShape(num_units, batch_size), 1, DataType::QSYMM16, sigmoi…  in validate()
884     const TensorInfo hidden_mul_res(TensorShape(num_units, batch_size), 1, DataType::S32);  in validate()
885 … const TensorInfo hidden_out_info(TensorShape(num_units, batch_size), 1, DataType::QASYMM8_SIGNED);  in validate()
895     const bool projection_tensor_copy_required = num_units != output_size;  in validate()
912 …const TensorInfo projection_weights_transposed(TensorShape(output_size, num_units), 1, lstm_params…  in validate()