Searched refs:worker_datasets (Results 1 – 2 of 2) sorted by relevance
405 def _create_iterators_per_worker(worker_datasets, input_workers, options=None): argument408 assert len(worker_datasets) == len(input_workers.worker_devices)414 worker_datasets[i], # pylint: disable=protected-access
1846 def _create_iterators_per_worker(worker_datasets, argument1852 assert len(worker_datasets) == len(input_workers.worker_devices)1858 dataset=worker_datasets[i],