Searched refs:GATHER_COLLECTIVE (Results 1 – 7 of 7) sorted by relevance
33 RingGatherer() : RingAlg(GATHER_COLLECTIVE, "Gather") {} in RingGatherer()
44 DCHECK_EQ(col_params->instance.type, GATHER_COLLECTIVE); in InitializeCollectiveParams()
241 col_params.instance.type == GATHER_COLLECTIVE || in ExecuteAsync()
183 col_params_.instance.type = GATHER_COLLECTIVE; in Init()538 cp.instance.type = GATHER_COLLECTIVE; in SetUpCollectiveParams()
621 } else if (cp->instance.type == GATHER_COLLECTIVE) { in AssignCollectiveType()
45 GATHER_COLLECTIVE, enumerator
75 col_params_.instance.type = GATHER_COLLECTIVE; in CollectiveGatherOpKernel()