Searched refs:CppCommHookInterface (Results 1 – 2 of 2) sorted by relevance
14 : public CppCommHookInterface<c10::intrusive_ptr<ProcessGroup>> {17 : CppCommHookInterface<c10::intrusive_ptr<ProcessGroup>>(state) {} in AllReduceCommHook()25 : public CppCommHookInterface<c10::intrusive_ptr<ProcessGroup>> {28 : CppCommHookInterface<c10::intrusive_ptr<ProcessGroup>>(state) {} in FP16CompressCommHook()41 : public CppCommHookInterface<c10::intrusive_ptr<ProcessGroup>> {45 : CppCommHookInterface<c10::intrusive_ptr<ProcessGroup>>(state) {} in _AllReduceBySumCommHook()
127 class CppCommHookInterface : public CommHookInterface { class129 explicit CppCommHookInterface(T state) : state_(std::move(state)) {} in CppCommHookInterface() function in c10d::CppCommHookInterface131 ~CppCommHookInterface() override = default;