Home
last modified time | relevance | path

Searched refs:expected_num_replicas_in_sync (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Done_device_strategy_test.py59 expected_num_replicas_in_sync=1,
69 expected_num_replicas_in_sync=1,
84 expected_num_replicas_in_sync=1,
H A Dstrategy_test_lib.py330 expected_num_replicas_in_sync, argument
342 self.assertEqual(expected_num_replicas_in_sync,
783 expected_num_replicas_in_sync=num_workers*num_gpus,
809 expected_num_replicas_in_sync=num_workers*num_gpus,
H A Dmirrored_strategy_test.py192 expected_num_replicas_in_sync=2,
203 expected_num_replicas_in_sync=2,
220 expected_num_replicas_in_sync=2,
1232 expected_num_replicas_in_sync=num_workers*num_gpus,
1257 expected_num_replicas_in_sync=num_workers*num_gpus,
H A Dcollective_all_reduce_strategy_test.py384 expected_num_replicas_in_sync=3*devices_per_worker,
529 expected_num_replicas_in_sync=required_gpus,
628 expected_num_replicas_in_sync=required_gpus,
H A Dparameter_server_strategy_test.py652 expected_num_replicas_in_sync=required_gpus,
683 expected_num_replicas_in_sync=required_gpus,
H A Dtpu_strategy_test.py1290 expected_num_replicas_in_sync=2,
1301 expected_num_replicas_in_sync=2,