Home
last modified time | relevance | path

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

/external/pytorch/test/distributed/
Dtest_c10d_pypg.py94 class AbstractDDPSingleRank(test_c10d_common.CommonDistributedDataParallelTest):
Dtest_c10d_common.py319 class CommonDistributedDataParallelTest: class
988 return CommonDistributedDataParallelTest.CustomOutput(o1=o1, o2=o2)
1785 instantiate_parametrized_tests(CommonDistributedDataParallelTest)
Dtest_c10d_ucc.py343 test_c10d_common.CommonDistributedDataParallelTest, MultiProcessTestCase
Dtest_c10d_gloo.py1485 test_c10d_common.CommonDistributedDataParallelTest, MultiProcessTestCase
Dtest_c10d_nccl.py845 test_c10d_common.CommonDistributedDataParallelTest, MultiProcessTestCase