Searched refs:_LegacyRebatchDataset (Results 1 – 5 of 5) sorted by relevance
408 rebatched_dataset = distribute._LegacyRebatchDataset(426 rebatched_dataset = distribute._LegacyRebatchDataset(439 rebatched_dataset = distribute._LegacyRebatchDataset(447 distribute._LegacyRebatchDataset(dataset.batch(4), num_replicas=4)450 distribute._LegacyRebatchDataset(dataset, num_replicas=4)458 rebatched_dataset = distribute._LegacyRebatchDataset(469 rebatched_dataset = distribute._LegacyRebatchDataset(480 rebatched_dataset = distribute._LegacyRebatchDataset(494 rebatched_dataset = distribute._LegacyRebatchDataset(510 rebatched_dataset = distribute._LegacyRebatchDataset([all …]
608 dataset = distribute._LegacyRebatchDataset(dataset, num_replicas=5)
38 return distribute._LegacyRebatchDataset(
226 class _LegacyRebatchDataset(dataset_ops.UnaryDataset): class290 super(_LegacyRebatchDataset, self).__init__(input_dataset, variant_tensor)
1071 return distribute._LegacyRebatchDataset(