Searched refs:class_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 | 624 def class_async_add(cls, to, x, y, z): member in AsyncExecutionClass
4179 self._test_async_function(AsyncExecutionClass.class_async_add)
4184 AsyncExecutionClass.class_async_add,
4197 ret += rref.rpc_sync().class_async_add(dst2, x, x, y)
4201 ret += rref.rpc_async().class_async_add(dst2, x, x, y).wait()
4205 ret += rref.remote().class_async_add(dst2, x, x, y).to_here()
|