Searched refs:_get_local_replica_id (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ | ||
| D | one_device_strategy.py | 474 def _get_local_replica_id(self, replica_id_in_sync_group): member in OneDeviceExtended |
| D | parameter_server_strategy.py | 701 def _get_local_replica_id(self, replica_id_in_sync_group): member in ParameterServerStrategyExtended |
| D | distribute_lib.py | 2928 self._local_replica_id = strategy.extended._get_local_replica_id( |
| D | collective_all_reduce_strategy.py | 933 def _get_local_replica_id(self, replica_id_in_sync_group): member in CollectiveAllReduceExtended |
| D | distribute_lib_test.py | 122 def _get_local_replica_id(self, replica_id_in_sync_group): member in _TestExtended |
| D | mirrored_strategy.py | 857 def _get_local_replica_id(self, replica_id_in_sync_group): member in MirroredExtended |
| D | tpu_strategy.py | 1523 def _get_local_replica_id(self, replica_id_in_sync_group): member in TPUExtended |