Searched refs:aux_input_to_forget_weights_ptr (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 55 const int8_t* aux_input_to_forget_weights_ptr, in ComputeRowSums() argument
82 tensor_utils::ReductionSumVector(aux_input_to_forget_weights_ptr, in ComputeRowSums()
814 const float* aux_input_to_forget_weights_ptr, in LstmStepFloat() argument
866 aux_input_to_forget_weights_ptr, output_state_ptr, in LstmStepFloat()
992 const int8_t* aux_input_to_forget_weights_ptr, in LstmStepHybrid() argument
1104 aux_input_to_input_weights_ptr, aux_input_to_forget_weights_ptr, in LstmStepHybrid()
1160 aux_input_zp, aux_input_to_forget_weights_ptr, in LstmStepHybrid()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 46 const float* aux_input_to_forget_weights_ptr, in LstmStepWithAuxInput() argument
138 aux_input_to_forget_weights_ptr, n_cell, n_aux_input, aux_input_ptr, in LstmStepWithAuxInput()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 163 const float* aux_input_to_forget_weights_ptr, in LstmStepCalibration() argument
218 aux_input_to_forget_weights_ptr, output_state_ptr, in LstmStepCalibration()
|