Searched defs:rref_t1 (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 | 107 def my_rref_add(rref_t1, t2): argument
123 def my_nested_rref_add(dst, rref_t1, t2): argument
166 def _run_trainer(rref_t1, t2, ps, rank_diff, sparse): argument
180 def _run_trainer_torchscript(rref_t1, t2, ps, rank_diff, sparse): argument
|