Searched refs:distributed_iterator (Results 1 – 2 of 2) sorted by relevance
661 distributed_iterator = iter(distributed_dataset)663 iterator_1 = distributed_iterator._values[0]669 iterator_2 = distributed_iterator._values[1]1492 distributed_iterator = iter(1494 worker_fn.get_concrete_function(distributed_iterator)1496 self.coordinator.schedule(worker_fn, args=(distributed_iterator,))
166 distributed_iterator = iter(170 def train_fn(distributed_iterator): argument176 optional_data = distributed_iterator.get_next_as_optional()181 train_fn(distributed_iterator)