Searched refs:PerWorkerDatasetFromDataset (Results 1 – 1 of 1) sorted by relevance
408 class PerWorkerDatasetFromDataset(PerWorkerDatasetFromDatasetFunction): class444 super(PerWorkerDatasetFromDataset, self).__init__(dataset_fn, coordinator)453 return PerWorkerDatasetFromDataset(dataset_or_dataset_fn, coordinator)