Searched refs:_get_replica_id_in_sync_group (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ | ||
| D | mirrored_run.py | 299 dist.extended._get_replica_id_in_sync_group(replica_id)) # pylint: disable=protected-access |
| D | parameter_server_strategy.py | 691 def _get_replica_id_in_sync_group(self, replica_id): member in ParameterServerStrategyExtended |
| D | collective_all_reduce_strategy.py | 985 def _get_replica_id_in_sync_group(self, replica_id): member in CollectiveAllReduceExtended |
| D | mirrored_strategy.py | 906 def _get_replica_id_in_sync_group(self, replica_id): member in MirroredExtended |
| D | distribute_lib.py | 3760 def _get_replica_id_in_sync_group(self, replica_id): member in _DefaultDistributionExtended |