_gpu_model_with_ddp_comm_hook (reference) in projects: aosp_15_r20

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 D	test_c10d_nccl.py	`1743 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 D	test_c10d_ucc.py	`814 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 D	test_c10d_gloo.py	`1986 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 D	test_c10d_common.py	`828 def _gpu_model_with_ddp_comm_hook( member in CommonDistributedDataParallelTest`