Searched refs:NcclBase (Results 1 – 5 of 5) sorted by relevance
26 NcclBase::NcclBase(CollectiveType type, const string& name) in NcclBase() function in tensorflow::NcclBase29 Status NcclBase::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()61 Status NcclBase::InitializeCollectiveContext( in InitializeCollectiveContext()70 Status NcclBase::InitializeCollectiveGroupRuntimeDetails( in InitializeCollectiveGroupRuntimeDetails()
23 class NcclBase : public CollectiveImplementationInterface {25 explicit NcclBase(CollectiveType type, const string& name);26 ~NcclBase() override = default;
23 class NcclBroadcaster : public NcclBase {25 NcclBroadcaster() : NcclBase(BROADCAST_COLLECTIVE, "NcclBroadcast") {} in NcclBroadcaster()
23 class NcclGatherer : public NcclBase {25 NcclGatherer() : NcclBase(GATHER_COLLECTIVE, "NcclGather") {} in NcclGatherer()
23 class NcclReducer : public NcclBase {25 NcclReducer() : NcclBase(REDUCTION_COLLECTIVE, "NcclReduce") {} in NcclReducer()