Searched refs:_test_DistributedDataParallel_with_amp (Results 1 – 1 of 1) sorted by relevance
5500 def _test_DistributedDataParallel_with_amp(self, grad_is_view=False): member in DistributedTest._DistTestBase5562 ddp_model_grad_not_view = self._test_DistributedDataParallel_with_amp(5565 ddp_model_grad_is_view = self._test_DistributedDataParallel_with_amp(