Searched refs:UpdateLstmCellFloat (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 100 void UpdateLstmCellFloat(int n_batch, int n_cell, float* cell_state, in UpdateLstmCellFloat() function
236 UpdateLstmCellFloat(n_batch, n_cell, cell_state_ptr, input_gate_scratch, in LstmStepCalibration()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 218 void UpdateLstmCellFloat(int n_batch, int n_cell, float* cell_state, in UpdateLstmCellFloat() function
882 UpdateLstmCellFloat(n_batch, n_cell, cell_state_ptr, input_gate_scratch, in LstmStepFloat()
1189 UpdateLstmCellFloat(n_batch, n_cell, cell_state_ptr, input_gate_scratch, in LstmStepHybrid()
|