Searched refs:multi_worker_iterator (Results 1 – 1 of 1) sorted by relevance
991 self, fn, multi_worker_iterator, iterations, initial_loop_values=None): argument1021 per_replica_inputs = multi_worker_iterator.get_next()