Home
last modified time | relevance | path

Searched defs:_get_process_group (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_pypg.py110 def _get_process_group(self): member in AbstractDDPSingleRank
H A Dtest_c10d_ucc.py349 def _get_process_group(self): member in DistributedDataParallelTest
H A Dtest_c10d_nccl.py854 def _get_process_group(self): member in DistributedDataParallelTest
2234 def _get_process_group(self): member in WorkHookTest
H A Dtest_c10d_gloo.py1491 def _get_process_group(self): member in DistributedDataParallelTest
H A Dtest_c10d_common.py391 def _get_process_group(self): member in CommonDistributedDataParallelTest