Searched defs:AllReduceCommHook (Results 1 – 1 of 1) sorted by relevance
13 class AllReduceCommHook class16 explicit AllReduceCommHook(const c10::intrusive_ptr<ProcessGroup>& state) in AllReduceCommHook() function in c10d::AllReduceCommHook