Searched refs:REDUCTION_COLLECTIVE (Results 1 – 11 of 11) sorted by relevance
32 if (col_params->instance.type != REDUCTION_COLLECTIVE || in InitializeCollectiveParams()36 REDUCTION_COLLECTIVE, "; or collective name ", in InitializeCollectiveParams()
123 col_params_.instance.type = REDUCTION_COLLECTIVE; in Init()
126 col_params_.instance.type = REDUCTION_COLLECTIVE; in CollectiveReduceOpKernel()
33 RingReducer() : RingAlg(REDUCTION_COLLECTIVE, "Reduce") {} in RingReducer()
91 cp.instance.type = REDUCTION_COLLECTIVE; in TEST_F()168 cp->instance.type = REDUCTION_COLLECTIVE; in TEST_F()
48 CHECK_EQ(col_params->instance.type, REDUCTION_COLLECTIVE); in InitializeCollectiveParams()
240 const Tensor* input = (col_params.instance.type == REDUCTION_COLLECTIVE || in ExecuteAsync()
615 } else if (cp->instance.type == REDUCTION_COLLECTIVE) { in AssignCollectiveType()687 bool nccl = cp->instance.type == REDUCTION_COLLECTIVE && in CompleteInstanceFromInitializedIRec()
205 col_params_.instance.type = REDUCTION_COLLECTIVE; in Init()565 cp.instance.type = REDUCTION_COLLECTIVE; in SetUpCollectiveParams()
43 REDUCTION_COLLECTIVE = 0, enumerator
197 cp.instance.type = REDUCTION_COLLECTIVE; in DefineCollectiveParams()