Home
last modified time | relevance | path

Searched refs:REDUCTION_COLLECTIVE (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer.cc32 if (col_params->instance.type != REDUCTION_COLLECTIVE || in InitializeCollectiveParams()
36 REDUCTION_COLLECTIVE, "; or collective name ", in InitializeCollectiveParams()
Dcollective_nccl_reducer_test.cc123 col_params_.instance.type = REDUCTION_COLLECTIVE; in Init()
Dcollective_ops.cc126 col_params_.instance.type = REDUCTION_COLLECTIVE; in CollectiveReduceOpKernel()
/external/tensorflow/tensorflow/core/common_runtime/
Dring_reducer.h33 RingReducer() : RingAlg(REDUCTION_COLLECTIVE, "Reduce") {} in RingReducer()
Dcollective_param_resolver_local_test.cc91 cp.instance.type = REDUCTION_COLLECTIVE; in TEST_F()
168 cp->instance.type = REDUCTION_COLLECTIVE; in TEST_F()
Dring_reducer.cc48 CHECK_EQ(col_params->instance.type, REDUCTION_COLLECTIVE); in InitializeCollectiveParams()
Dbase_collective_executor.cc240 const Tensor* input = (col_params.instance.type == REDUCTION_COLLECTIVE || in ExecuteAsync()
Dcollective_param_resolver_local.cc615 } else if (cp->instance.type == REDUCTION_COLLECTIVE) { in AssignCollectiveType()
687 bool nccl = cp->instance.type == REDUCTION_COLLECTIVE && in CompleteInstanceFromInitializedIRec()
Dring_reducer_test.cc205 col_params_.instance.type = REDUCTION_COLLECTIVE; in Init()
565 cp.instance.type = REDUCTION_COLLECTIVE; in SetUpCollectiveParams()
/external/tensorflow/tensorflow/core/framework/
Dcollective.h43 REDUCTION_COLLECTIVE = 0, enumerator
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed_test.cc197 cp.instance.type = REDUCTION_COLLECTIVE; in DefineCollectiveParams()