Searched refs:forget_layer_norm_coefficients_ptr (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 57 const float* forget_layer_norm_coefficients_ptr, in LstmStepWithAuxInput() argument
74 const bool use_layer_norm = (forget_layer_norm_coefficients_ptr != nullptr); in LstmStepWithAuxInput()
230 forget_layer_norm_coefficients_ptr, n_cell, forget_gate_scratch, in LstmStepWithAuxInput()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 174 const float* forget_layer_norm_coefficients_ptr, in LstmStepCalibration() argument
220 cell_to_forget_weights_ptr, forget_layer_norm_coefficients_ptr, in LstmStepCalibration()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 825 const float* forget_layer_norm_coefficients_ptr, in LstmStepFloat() argument
868 cell_to_forget_weights_ptr, forget_layer_norm_coefficients_ptr, in LstmStepFloat()
1016 const float* forget_layer_norm_coefficients_ptr, in LstmStepHybrid()
1166 forget_layer_norm_coefficients_ptr, forget_gate_bias_ptr, n_batch, in LstmStepHybrid()
|