Home
last modified time | relevance | path

Searched refs:_gpu_model_with_ddp_comm_hook (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_nccl.py1743 def _gpu_model_with_ddp_comm_hook( member in DistributedDataParallelTest
1776 gpu_model = self._gpu_model_with_ddp_comm_hook(process_group, self._simple_hook)
1804 gpu_model = self._gpu_model_with_ddp_comm_hook(
1832 gpu_model = self._gpu_model_with_ddp_comm_hook(
1853 gpu_model = self._gpu_model_with_ddp_comm_hook(
1877 gpu_model = self._gpu_model_with_ddp_comm_hook(
1909 gpu_model = self._gpu_model_with_ddp_comm_hook(
2036 gpu_model = self._gpu_model_with_ddp_comm_hook(
H A Dtest_c10d_ucc.py814 def _gpu_model_with_ddp_comm_hook( member in DistributedDataParallelTest
841 gpu_model = self._gpu_model_with_ddp_comm_hook(process_group, self._simple_hook)
H A Dtest_c10d_gloo.py1986 def _gpu_model_with_ddp_comm_hook( member in DistributedDataParallelTest
2013 gpu_model = self._gpu_model_with_ddp_comm_hook(process_group, self._simple_hook)
H A Dtest_c10d_common.py828 def _gpu_model_with_ddp_comm_hook( member in CommonDistributedDataParallelTest