Searched refs:my_rref_function (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/ |
| H A D | rpc_test.py | 276 def my_rref_function(rref_a, rref_b): function
972 worker_name(dst_rank), my_rref_function, args=(rref_a, rref_b)
987 worker_name(user_rank), my_rref_function, args=(rref_a, rref_b)
1002 worker_name(dst_rank), my_rref_function, args=(rref_a, rref_b)
3068 worker_name(dst_rank), my_rref_function, args=(rref_a, rref_b)
|