Searched refs:worker_datasets (Results 1 – 1 of 1) sorted by relevance
1891 def _create_iterators_per_worker(worker_datasets, argument1897 assert len(worker_datasets) == len(input_workers.worker_devices)1904 dataset=worker_datasets[i],1909 iterator = _SingleWorkerDatasetIterator(worker_datasets[i], worker,