Searched refs:context_cleanup_test_helper (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/ |
| H A D | dist_autograd_test.py | 572 def context_cleanup_test_helper(self, rpc_args, func, nested=False): member in CommonDistAutogradTest
1067 self.context_cleanup_test_helper(rpc_args=(t1, t2), func=torch.add)
1072 self.context_cleanup_test_helper(rpc_args=(t1, t1), func=torch.add)
1080 self.context_cleanup_test_helper(
1398 self.context_cleanup_test_helper(rpc_args=(t1, t2), func=torch.add)
1403 self.context_cleanup_test_helper(rpc_args=(t1, t1), func=torch.add)
1407 self.context_cleanup_test_helper(rpc_args=(1, 1), func=my_scalar_add)
1415 self.context_cleanup_test_helper(
2598 def context_cleanup_test_helper(self, rpc_args, func): member in FaultyAgentDistAutogradTest
2630 self.context_cleanup_test_helper(rpc_args=(t1, t2), func=torch.add)
|