Searched refs:_num_replicas_in_sync (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy.py | 346 num_replicas_in_sync=self._num_replicas_in_sync,
354 num_replicas_in_sync=self._num_replicas_in_sync)
372 num_replicas_in_sync=self._num_replicas_in_sync)
394 num_replicas_in_sync=self._num_replicas_in_sync)
404 for replica_id in range(self._num_replicas_in_sync):
407 self._num_replicas_in_sync)))
427 if self._num_replicas_in_sync > 1:
525 reduce_op, value, destinations, self._num_replicas_in_sync)
643 def _num_replicas_in_sync(self): member in ParameterServerStrategyExtended
|
| D | tpu_strategy.py | 884 num_replicas_in_sync=self._num_replicas_in_sync)
898 num_replicas_in_sync=self._num_replicas_in_sync))
948 num_replicas_in_sync=self._num_replicas_in_sync,
966 num_replicas_in_sync=self._num_replicas_in_sync))
982 for replica_id in range(self._num_replicas_in_sync):
985 self._num_replicas_in_sync)))
1024 for replica_id in range(self._num_replicas_in_sync):
1048 initial_loop_values = initial_loop_values * self._num_replicas_in_sync
1076 output_num = len(last_step_tensor_outputs) // self._num_replicas_in_sync
1236 value = math_ops.scalar_mul((1./self._num_replicas_in_sync), value)
[all …]
|
| D | mirrored_strategy.py | 541 num_replicas_in_sync=self._num_replicas_in_sync)
553 num_replicas_in_sync=self._num_replicas_in_sync))
570 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)))
736 reduce_op, value, destinations, self._num_replicas_in_sync)
845 def _num_replicas_in_sync(self): member in MirroredExtended
|
| D | parameter_server_strategy_v2.py | 583 def _num_replicas_in_sync(self): member in ParameterServerStrategyV2Extended
598 if self._num_replicas_in_sync > 1:
817 num_replicas_in_sync=self._num_replicas_in_sync,
830 num_replicas_in_sync=self._num_replicas_in_sync)
|
| D | distribute_lib.py | 500 self._num_replicas_in_sync = num_replicas_in_sync
505 return self._num_replicas_in_sync
531 if global_batch_size % self._num_replicas_in_sync != 0:
534 (global_batch_size, self._num_replicas_in_sync))
535 return global_batch_size // self._num_replicas_in_sync
590 self._num_replicas_in_sync = num_replicas_in_sync
595 return self._num_replicas_in_sync
1565 return self._extended._num_replicas_in_sync # pylint: disable=protected-access
2688 def _num_replicas_in_sync(self): member in StrategyExtendedV2
3723 def _num_replicas_in_sync(self): member in _DefaultDistributionExtended
|
| D | collective_all_reduce_strategy.py | 607 num_replicas_in_sync=self._num_replicas_in_sync)
624 num_replicas_in_sync=self._num_replicas_in_sync,
651 replica_id, self._num_replicas_in_sync)
662 num_replicas_in_sync=self._num_replicas_in_sync,
932 def _num_replicas_in_sync(self): member in CollectiveAllReduceExtended
|
| D | one_device_strategy.py | 435 def _num_replicas_in_sync(self): member in OneDeviceExtended
|
| D | collective_all_reduce_strategy_test.py | 573 self.assertAllEqual(list(range(strategy.extended._num_replicas_in_sync)),
|
| D | input_lib.py | 1205 self._num_replicas_in_sync = num_replicas_in_sync
1239 self._num_replicas_in_sync)
|