Searched refs:NcclReducer (Results 1 – 3 of 3) sorted by relevance
23 class NcclReducer : public NcclBase {25 NcclReducer() : NcclBase(REDUCTION_COLLECTIVE, "NcclReduce") {} in NcclReducer() function26 ~NcclReducer() override = default;
26 void NcclReducer::Run(StatusCallback done) { in Run()124 REGISTER_COLLECTIVE(NcclReduce, NcclReducer);
328 auto* reducer = new NcclReducer(); in RunReduce()