_test_ddp_with_process_group (reference) in projects: aosp_15_r20

Searched refs:_test_ddp_with_process_group (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/
H A D	test_c10d_pypg.py	`135 self._test_ddp_with_process_group(pg, [torch.device("cpu")], device_ids=None) 140 self._test_ddp_with_process_group(`
H A D	test_c10d_ucc.py	`360 self._test_ddp_with_process_group(`
H A D	test_c10d_common.py	`748 def _test_ddp_with_process_group( member in CommonDistributedDataParallelTest`
H A D	test_c10d_gloo.py	`1509 self._test_ddp_with_process_group(`
H A D	test_c10d_nccl.py	`865 self._test_ddp_with_process_group(`