Searched refs:WorkerSpecificIterableDataset (Results 1 – 1 of 1) sorted by relevance
741 class WorkerSpecificIterableDataset(IterableDataset): class1656 dataset = WorkerSpecificIterableDataset(sizes_for_all_workers)1731 dataset = WorkerSpecificIterableDataset(sizes_for_all_workers)1784 dataset = WorkerSpecificIterableDataset(sizes_for_all_workers)