Searched refs:funcol_py (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/debug/ |
| H A D | _comm_mode.py | 27 funcol_py = torch.ops.c10d_functional variable
32 funcol_native.all_gather_into_tensor: funcol_py.all_gather_into_tensor,
33 funcol_native.all_gather_into_tensor_coalesced: funcol_py.all_gather_into_tensor_coalesced,
34 funcol_native.all_reduce: funcol_py.all_reduce,
35 funcol_native.all_reduce_coalesced: funcol_py.all_reduce_coalesced,
36 funcol_native.all_to_all_single: funcol_py.all_to_all_single,
37 funcol_native.broadcast: funcol_py.broadcast,
38 funcol_native.reduce_scatter_tensor: funcol_py.reduce_scatter_tensor,
39 funcol_native.reduce_scatter_tensor_coalesced: funcol_py.reduce_scatter_tensor_coalesced,
41 funcol_autograd.all_to_all_single: funcol_py.all_to_all_single,
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/experimental/ |
| H A D | test_local_map.py | 23 funcol_py = torch.ops.c10d_functional variable
165 comm_mode.get_comm_counts()[funcol_py.all_gather_into_tensor], 1
|