Searched refs:static_async_add (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/ |
| H A D | rpc_test.py | 617 def static_async_add(to, x, y, z): member in AsyncExecutionClass
4168 self._test_async_function(AsyncExecutionClass.static_async_add)
4173 AsyncExecutionClass.static_async_add,
4196 ret = rref.rpc_sync().static_async_add(dst2, x, x, y)
4200 ret = rref.rpc_async().static_async_add(dst2, x, x, y).wait()
4204 ret = rref.remote().static_async_add(dst2, x, x, y).to_here()
|