Searched refs:_gpu_model_with_ddp_comm_hook (Results 1 – 4 of 4) sorted by relevance
1743 def _gpu_model_with_ddp_comm_hook( member in DistributedDataParallelTest1776 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(
814 def _gpu_model_with_ddp_comm_hook( member in DistributedDataParallelTest841 gpu_model = self._gpu_model_with_ddp_comm_hook(process_group, self._simple_hook)
1986 def _gpu_model_with_ddp_comm_hook( member in DistributedDataParallelTest2013 gpu_model = self._gpu_model_with_ddp_comm_hook(process_group, self._simple_hook)
828 def _gpu_model_with_ddp_comm_hook( member in CommonDistributedDataParallelTest