Searched refs:_num_replicas_in_sync (Results 1 – 6 of 6) sorted by relevance
237 self._tpu_devices = self._tpu_devices[:self._num_replicas_in_sync]260 self._num_replicas_in_sync)272 num_replicas_in_sync=self._num_replicas_in_sync))325 for replica_id in range(self._num_replicas_in_sync):344 initial_loop_values = initial_loop_values * self._num_replicas_in_sync372 output_num = len(last_step_tensor_outputs) // self._num_replicas_in_sync445 value *= (1. / self._num_replicas_in_sync)551 def _num_replicas_in_sync(self): member in TPUExtended
252 self._num_replicas_in_sync = num_replicas_in_sync257 return self._num_replicas_in_sync283 if global_batch_size % self._num_replicas_in_sync != 0:286 (global_batch_size, self._num_replicas_in_sync))287 return global_batch_size // self._num_replicas_in_sync564 return self._extended._num_replicas_in_sync # pylint: disable=protected-access1292 def _num_replicas_in_sync(self): member in DistributionStrategyExtended1635 def _num_replicas_in_sync(self): member in _DefaultDistributionExtended
257 self._num_replicas_in_sync)275 num_replicas_in_sync=self._num_replicas_in_sync)303 if self._num_replicas_in_sync > 1:513 def _num_replicas_in_sync(self): member in ParameterServerStrategyExtended
333 self._num_replicas_in_sync)348 num_replicas_in_sync=self._num_replicas_in_sync)470 def _num_replicas_in_sync(self): member in CollectiveAllReduceExtended
571 dataset, self._input_workers, self._num_replicas_in_sync)583 num_replicas_in_sync=self._num_replicas_in_sync))753 def _num_replicas_in_sync(self): member in MirroredExtended
175 def _num_replicas_in_sync(self): member in OneDeviceExtended