Searched refs:_make_rebatch_fn (Results 1 – 1 of 1) sorted by relevance
1003 rebatch_fn = self._make_rebatch_fn(dataset, num_workers,1040 def _make_rebatch_fn(self, dataset, num_workers, num_replicas_in_sync): member in DistributedDataset