Searched refs:num_replicas_per_worker (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | rebatch_dataset_test.py | 41 def _test(self, global_batch_size, num_workers, num_replicas_per_worker, argument
55 num_replicas_per_worker, i)))
58 self.assertLen(batch_sizes, num_workers * num_replicas_per_worker)
65 offset = step_index * num_replicas_per_worker
68 num_replicas_per_worker])
86 num_replicas_per_worker = 2
90 num_replicas_per_worker,
94 self._test(global_batch_size, num_workers, num_replicas_per_worker,
104 num_replicas_per_worker = 1
109 num_replicas_per_worker,
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | distribute.py | 354 num_replicas_per_worker, worker_index): argument
438 num_subbatches = num_workers * num_replicas_per_worker
440 offset = worker_index * num_replicas_per_worker
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 1057 num_replicas_per_worker = num_replicas_in_sync // num_workers
1066 batch_size, num_workers, num_replicas_per_worker, worker_index)
1068 dataset, batch_sizes).prefetch(num_replicas_per_worker)
1072 dataset, num_replicas_in_sync).prefetch(num_replicas_per_worker)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training.py | 2824 num_replicas_per_worker = len(strategy.extended.worker_devices)
2825 for replica_id in range(num_replicas_per_worker):
2826 ordered_replicas += replicas[replica_id::num_replicas_per_worker]
|