Searched refs:test_num_replicas_in_sync (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ | ||
| H A D | collective_all_reduce_strategy_test.py | 297 def test_num_replicas_in_sync(self): member in DistributedCollectiveAllReduceStrategyTest |
| H A D | parameter_server_strategy_test.py | 577 def test_num_replicas_in_sync(self): member in ParameterServerStrategyTest |
| H A D | tpu_strategy_test.py | 1310 def test_num_replicas_in_sync(self): member in TPUStrategyDistributionTest |
| H A D | mirrored_strategy_test.py | 1198 def test_num_replicas_in_sync(self, distribution): member in MultiWorkerMirroredStrategyTest |