Searched refs:_num_replicas_in_sync (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy.py | 346 num_replicas_in_sync=self._num_replicas_in_sync)
353 num_replicas_in_sync=self._num_replicas_in_sync)
371 num_replicas_in_sync=self._num_replicas_in_sync)
393 num_replicas_in_sync=self._num_replicas_in_sync)
403 for replica_id in range(self._num_replicas_in_sync):
406 self._num_replicas_in_sync)))
428 if self._num_replicas_in_sync > 1:
520 reduce_op, value, destinations, self._num_replicas_in_sync)
654 def _num_replicas_in_sync(self): member in ParameterServerStrategyExtended
|
| D | tpu_strategy.py | 885 num_replicas_in_sync=self._num_replicas_in_sync)
899 num_replicas_in_sync=self._num_replicas_in_sync))
949 num_replicas_in_sync=self._num_replicas_in_sync)
966 num_replicas_in_sync=self._num_replicas_in_sync))
981 for replica_id in range(self._num_replicas_in_sync):
984 self._num_replicas_in_sync)))
1023 for replica_id in range(self._num_replicas_in_sync):
1047 initial_loop_values = initial_loop_values * self._num_replicas_in_sync
1075 output_num = len(last_step_tensor_outputs) // self._num_replicas_in_sync
1221 value *= (1. / self._num_replicas_in_sync)
[all …]
|
| D | mirrored_strategy.py | 542 num_replicas_in_sync=self._num_replicas_in_sync)
554 num_replicas_in_sync=self._num_replicas_in_sync))
571 num_replicas_in_sync=self._num_replicas_in_sync)
585 num_replicas_in_sync=self._num_replicas_in_sync))
593 for replica_id in range(self._num_replicas_in_sync):
596 self._num_replicas_in_sync)))
730 reduce_op, value, destinations, self._num_replicas_in_sync)
805 def _num_replicas_in_sync(self): member in MirroredExtended
|
| D | distribute_lib.py | 483 self._num_replicas_in_sync = num_replicas_in_sync
488 return self._num_replicas_in_sync
514 if global_batch_size % self._num_replicas_in_sync != 0:
517 (global_batch_size, self._num_replicas_in_sync))
518 return global_batch_size // self._num_replicas_in_sync
573 self._num_replicas_in_sync = num_replicas_in_sync
578 return self._num_replicas_in_sync
1516 return self._extended._num_replicas_in_sync # pylint: disable=protected-access
2579 def _num_replicas_in_sync(self): member in StrategyExtendedV2
3499 def _num_replicas_in_sync(self): member in _DefaultDistributionExtended
|
| D | collective_all_reduce_strategy.py | 590 num_replicas_in_sync=self._num_replicas_in_sync)
606 num_replicas_in_sync=self._num_replicas_in_sync,
631 replica_id, self._num_replicas_in_sync)
642 num_replicas_in_sync=self._num_replicas_in_sync,
915 def _num_replicas_in_sync(self): member in CollectiveAllReduceExtended
|
| D | one_device_strategy.py | 433 def _num_replicas_in_sync(self): member in OneDeviceExtended
|
| D | collective_all_reduce_strategy_test.py | 566 self.assertAllEqual(list(range(strategy.extended._num_replicas_in_sync)),
|