Searched refs:state_update_fn (Results 1 – 1 of 1) sorted by relevance
635 times=times, state=state, state_update_fn=_batch_loss_filtering_step,672 state_update_fn=_call_prediction_step,692 def _state_update_loop(self, times, state, state_update_fn, outputs): argument750 step_state, output_values = state_update_fn(