Home
last modified time | relevance | path

Searched refs:multi_worker_iterator (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy.py285 self, fn, multi_worker_iterator, iterations, initial_loop_values=None): argument
286 output_shapes = multi_worker_iterator.output_shapes
323 per_replica_inputs = multi_worker_iterator.get_next()