Searched refs:rpc_async_method (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/ |
| H A D | dist_optimizer_test.py | 77 def rpc_async_method(method, obj_rref, *args, **kwargs): function
118 output1 = rpc_async_method(MyModule.forward, remote_module1, t2)
119 output2 = rpc_async_method(MyModule.forward, remote_module2, output1.wait())
186 output1 = rpc_async_method(MyModule.forward, remote_module1, t2)
187 output2 = rpc_async_method(MyModule.forward, remote_module2, output1.wait())
193 new_w1 = rpc_async_method(MyModule.get_w, remote_module1).wait()
194 new_w2 = rpc_async_method(MyModule.get_w, remote_module2).wait()
|