HomeSort by relevance Sort by last modified time
    Searched refs:REDUCTION_COLLECTIVE (Results 1 - 11 of 11) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/
ring_reducer.h 33 RingReducer() : RingAlg(REDUCTION_COLLECTIVE, "Reduce") {}
collective_param_resolver_local_test.cc 91 cp.instance.type = REDUCTION_COLLECTIVE;
168 cp->instance.type = REDUCTION_COLLECTIVE;
ring_reducer.cc 48 CHECK_EQ(col_params->instance.type, REDUCTION_COLLECTIVE);
base_collective_executor.cc 240 const Tensor* input = (col_params.instance.type == REDUCTION_COLLECTIVE ||
collective_param_resolver_local.cc 615 } else if (cp->instance.type == REDUCTION_COLLECTIVE) {
687 bool nccl = cp->instance.type == REDUCTION_COLLECTIVE &&
ring_reducer_test.cc 205 col_params_.instance.type = REDUCTION_COLLECTIVE;
565 cp.instance.type = REDUCTION_COLLECTIVE;
  /external/tensorflow/tensorflow/core/kernels/
collective_nccl_reducer.cc 32 if (col_params->instance.type != REDUCTION_COLLECTIVE ||
36 REDUCTION_COLLECTIVE, "; or collective name ",
collective_nccl_reducer_test.cc 123 col_params_.instance.type = REDUCTION_COLLECTIVE;
collective_ops.cc 126 col_params_.instance.type = REDUCTION_COLLECTIVE;
  /external/tensorflow/tensorflow/core/framework/
collective.h 43 REDUCTION_COLLECTIVE = 0,
  /external/tensorflow/tensorflow/core/distributed_runtime/
collective_param_resolver_distributed_test.cc 197 cp.instance.type = REDUCTION_COLLECTIVE;

Completed in 246 milliseconds