Searched refs:input_layer_norm_coefficients_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 | 56 const float* input_layer_norm_coefficients_ptr, in LstmStepWithAuxInput() argument 216 input_layer_norm_coefficients_ptr, n_cell, input_gate_scratch, in LstmStepWithAuxInput()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
H A D | lstm.cc | 174 const float* input_layer_norm_coefficients_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 | 867 const float* input_layer_norm_coefficients_ptr, in LstmStepFloat() argument 905 input_layer_norm_coefficients_ptr, in LstmStepFloat() 1071 const float* input_layer_norm_coefficients_ptr, in LstmStepHybrid() 1205 input_layer_norm_coefficients_ptr, input_gate_bias_ptr, n_batch, in LstmStepHybrid()
|