Searched defs:_num_replicas_in_sync (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ | ||
| H A D | parameter_server_strategy_v2.py | 601 def _num_replicas_in_sync(self): member in ParameterServerStrategyV2Extended |
| H A D | collective_all_reduce_strategy.py | 970 def _num_replicas_in_sync(self): member in CollectiveAllReduceExtended |
| H A D | one_device_strategy.py | 433 def _num_replicas_in_sync(self): member in OneDeviceExtended |
| H A D | parameter_server_strategy.py | 641 def _num_replicas_in_sync(self): member in ParameterServerStrategyExtended |
| H A D | mirrored_strategy.py | 851 def _num_replicas_in_sync(self): member in MirroredExtended |
| H A D | distribute_lib.py | 2695 def _num_replicas_in_sync(self): member in StrategyExtendedV2 |
| H A D | tpu_strategy.py | 1483 def _num_replicas_in_sync(self): member in TPUExtended |