Searched refs:udf_with_torch_ops (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/ |
| H A D | rpc_test.py | 64 def udf_with_torch_ops(device=-1, use_record_function=False): function
1805 fut = rpc.rpc_async(dst_worker, udf_with_torch_ops, args=())
1815 fut = rpc.rpc_async(dst_worker, udf_with_torch_ops, args=())
1829 fut = rpc.rpc_async(dst_worker, udf_with_torch_ops, args=())
1856 fut = rpc.rpc_async(dst_worker, udf_with_torch_ops, args=())
1865 udf_with_torch_ops.__qualname__,
1922 fut = rpc.rpc_async(dst_worker, udf_with_torch_ops, args=())
1931 udf_with_torch_ops,
1939 udf_with_torch_ops.__qualname__,
2060 fut = rpc.rpc_async(dst_worker, udf_with_torch_ops, args=(-1, True))
[all …]
|