Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer.h25 NcclReducer() : NcclBase(REDUCTION_COLLECTIVE, "NcclReduce") {} in NcclReducer()
Dcollective_nccl.cc38 case REDUCTION_COLLECTIVE: in InitializeCollectiveParams()
Dcollective_ops.cc213 col_params_->instance.type = REDUCTION_COLLECTIVE; in CollectiveReduceOpKernel()
527 col_params->instance.type = REDUCTION_COLLECTIVE; in ComputeAsync()
Dcollective_nccl_test.cc449 : NcclTestBase(/*collective_type=*/REDUCTION_COLLECTIVE, in NcclReducerTest()
/external/tensorflow/tensorflow/core/common_runtime/
Dring_reducer.h33 RingReducer() : RingAlg(REDUCTION_COLLECTIVE, "Reduce") {} in RingReducer()
Dbase_collective_executor.cc300 const Tensor* input = (col_params->instance.type == REDUCTION_COLLECTIVE || in ExecuteAsync()
410 col_params.instance.type == REDUCTION_COLLECTIVE) { in CreateCollective()
Dring_reducer.cc50 CHECK_EQ(col_params->instance.type, REDUCTION_COLLECTIVE); in InitializeCollectiveParams()
Dring_reducer_test.cc217 col_params_->instance.type = REDUCTION_COLLECTIVE; in Init()
580 cp->instance.type = REDUCTION_COLLECTIVE; in SetUpCollectiveParams()
Dcollective_param_resolver_local_test.cc175 cp->instance.type = REDUCTION_COLLECTIVE; in TEST_F()
Dcollective_param_resolver_local.cc64 case REDUCTION_COLLECTIVE: in GetCollectiveName()
/external/tensorflow/tensorflow/core/nccl/
Dcollective_communicator.cc136 case REDUCTION_COLLECTIVE: { in Enqueue()
/external/tensorflow/tensorflow/core/framework/
Dcollective.h44 REDUCTION_COLLECTIVE = 0, enumerator
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed_test.cc201 cp->instance.type = REDUCTION_COLLECTIVE; in CreateCollectiveParams()