Searched refs:cell_gate_bias_ptr (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 829 const float* cell_gate_bias_ptr, const float* output_gate_bias_ptr, in LstmStepFloat() argument
877 cell_layer_norm_coefficients_ptr, cell_gate_bias_ptr, in LstmStepFloat()
1020 const float* cell_gate_bias_ptr, const float* output_gate_bias_ptr, in LstmStepHybrid()
1183 cell_gate_bias_ptr, n_batch, n_input, n_aux_input, n_output, n_cell, in LstmStepHybrid()
1362 const int32_t* cell_gate_bias_ptr, const int32_t* output_gate_bias_ptr, in LstmStepInteger8x8_16()
1442 cell_gate_bias_ptr, layer_norm_cell_scale_a, layer_norm_cell_scale_b, in LstmStepInteger8x8_16()
1614 const int32_t* cell_gate_bias_ptr, const int32_t* output_gate_bias_ptr, in LstmStepInteger8x8_8()
1654 layer_norm_cell_scale_b, cell_gate_bias_ptr, n_batch, n_input, n_output, in LstmStepInteger8x8_8()
|
| D | lstm.cc | 564 int32_t* cell_gate_bias_ptr = nullptr; in PopulateQuantizedLstmParams8x8_8() local
664 cell_gate_bias_ptr = cell_gate_bias->data.i32; in PopulateQuantizedLstmParams8x8_8()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 178 const float* cell_gate_bias_ptr, const float* output_gate_bias_ptr, in LstmStepCalibration() argument
230 cell_layer_norm_coefficients_ptr, cell_gate_bias_ptr, in LstmStepCalibration()
|