Searched defs:local_grads (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/testing/_internal/distributed/rpc/ |
| D | dist_autograd_test.py | 250 def _verify_backwards(self, exec_mode, tensors, context_id, local_grads, *args): argument
257 def _verify_backwards_remote(self, tensors, context_id, local_grads, *args): argument
645 def _backward_rref(self, callee, rref_owner, t1, t2, local_grads, sparse): argument
750 def _backward_multiple_round_trips(self, t1, t2, t3, t4, t5, local_grads, sparse): argument
929 def _backward_simple(self, dst, t1, t2, local_grads, sparse): argument
|