Searched defs:_prepare_single_device_module (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/ | ||
| H A D | test_c10d_common.py | 333 def _prepare_single_device_module( member in CommonDistributedDataParallelTest |
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ | ||
| H A D | distributed_test.py | 5220 def _prepare_single_device_module( member in DistributedTest._DistTestBase |