Home
last modified time | relevance | path

Searched refs:resolve_process_group (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DFunctional.cpp113 auto group = c10d::resolve_process_group(group_name); in all_reduce_()
136 auto group = c10d::resolve_process_group(group_name); in all_reduce_coalesced_()
179 auto group = c10d::resolve_process_group(group_name); in all_gather_into_tensor_coalesced()
203 auto group = c10d::resolve_process_group(group_name); in all_gather_into_tensor_out()
239 auto group = c10d::resolve_process_group(group_name); in reduce_scatter_tensor_coalesced()
268 auto group = c10d::resolve_process_group(group_name); in all_to_all_single()
285 auto group = c10d::resolve_process_group(group_name); in broadcast_()
608 auto group = c10d::resolve_process_group(group_name); in shard_dim_alltoall()
H A DGroupRegistry.hpp15 C10_EXPORT c10::intrusive_ptr<c10d::ProcessGroup> resolve_process_group(
H A DGroupRegistry.cpp82 c10::intrusive_ptr<c10d::ProcessGroup> resolve_process_group( in resolve_process_group() function
H A Dinit.cpp921 return ::c10d::resolve_process_group(group_name); in c10d_init()