Searched refs:cell_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 | 53 const float* cell_to_input_weights_ptr, in LstmStepWithAuxInput() argument
206 cell_to_input_weights_ptr, n_cell, cell_state_ptr, n_batch, in LstmStepWithAuxInput()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 171 const float* cell_to_input_weights_ptr, in LstmStepCalibration() argument
210 cell_to_input_weights_ptr, input_layer_norm_coefficients_ptr, in LstmStepCalibration()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_eval.cc | 864 const float* cell_to_input_weights_ptr, in LstmStepFloat() argument
904 cell_state_ptr, cell_to_input_weights_ptr, in LstmStepFloat()
1066 const int8_t* cell_to_input_weights_ptr, float cell_to_input_weights_scale, in LstmStepHybrid()
1204 cell_state_ptr, cell_to_input_weights_ptr, cell_to_input_weights_scale, in LstmStepHybrid()
|