ALLREDUCE (definition) in projects: aosp_15_r20

Searched defs:ALLREDUCE (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A D	default_comm_hooks.hpp	`9 ALLREDUCE = 1, enumerator`
H A D	Work.hpp	`16 ALLREDUCE = 1, enumerator`
/aosp_15_r20/external/pytorch/torch/distributed/algorithms/ddp_comm_hooks/
H A D	__init__.py	`54 ALLREDUCE = partial(_ddp_comm_hook_wrapper, comm_hook=default.allreduce_hook) variable in DDPCommHookType`