Searched refs:NcclReducer (Results 1 – 3 of 3) sorted by relevance
29 NcclReducer::NcclReducer() : col_ctx_(nullptr), col_params_(nullptr) {} in NcclReducer() function in tensorflow::NcclReducer31 Status NcclReducer::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()44 Status NcclReducer::InitializeCollectiveContext(CollectiveContext* col_ctx) { in InitializeCollectiveContext()52 Status NcclReducer::InitializeInstanceBeforeGroupDiscovery( in InitializeInstanceBeforeGroupDiscovery()74 void NcclReducer::Run(StatusCallback done) { in Run()202 REGISTER_COLLECTIVE(NcclReduce, NcclReducer);
23 class NcclReducer : public CollectiveImplementationInterface {25 NcclReducer();26 ~NcclReducer() override = default;
290 NcclReducer reducer; in DoReduce()