Searched refs:state_accumulators (Results 1 – 1 of 1) sorted by relevance
740 step_number, state, state_accumulators, output_accumulators, argument762 assert len(state_accumulators) == len(flat_step_state)766 flat_step_state, state_accumulators, nest.flatten(state)):