Searched refs:recurrent_to_input_weights_ptr (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 49 const float* recurrent_to_input_weights_ptr, in LstmStepWithAuxInput() argument
154 recurrent_to_input_weights_ptr, n_cell, n_output, output_state_ptr, in LstmStepWithAuxInput()
171 recurrent_to_input_weights_ptr, n_cell, n_output, output_state_ptr, in LstmStepWithAuxInput()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_eval.cc | 89 const int8_t* recurrent_to_input_weights_ptr, in ComputeRowSums() argument
124 tensor_utils::ReductionSumVector(recurrent_to_input_weights_ptr, in ComputeRowSums()
860 const float* recurrent_to_input_weights_ptr, in LstmStepFloat() argument
902 recurrent_to_input_weights_ptr, in LstmStepFloat()
1054 const int8_t* recurrent_to_input_weights_ptr, in LstmStepHybrid() argument
1162 recurrent_to_input_weights_ptr, recurrent_to_forget_weights_ptr, in LstmStepHybrid()
1202 recurrent_to_input_weights_ptr, recurrent_to_input_weights_ledger_ptr, in LstmStepHybrid()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 167 const float* recurrent_to_input_weights_ptr, in LstmStepCalibration() argument
209 recurrent_to_input_weights_ptr, cell_state_ptr, in LstmStepCalibration()
|