Searched refs:my_py_nested_call (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/ |
| H A D | dist_autograd_test.py | 131 def my_py_nested_call(t1, t2, dst, world_size, hops): function
136 my_py_nested_call,
350 my_py_nested_call,
356 my_py_nested_call,
432 my_py_nested_call,
444 my_py_nested_call,
1081 rpc_args=args, func=my_py_nested_call, nested=True
1416 rpc_args=args, func=my_py_nested_call, nested=True
1920 my_py_nested_call,
|