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

  /external/tensorflow/tensorflow/core/nccl/
nccl_manager_test.cc 84 } else if (reduction_op == ncclSum) {
110 } else if (reduction_op == ncclSum) {
307 /*num_nodes=*/1, num_ranks, ncclSum,
351 ncclSum);
469 this->MakeReductionTestCase(1 /* num_nodes */, num_ranks, ncclSum,
485 ncclSum);
504 this->MakeReductionTestCase(1 /* num_nodes */, num_ranks, ncclSum,
520 ncclSum);
532 this->MakeReductionTestCase(1 /* num_nodes */, num_ranks, ncclSum,
548 ncclSum);
    [all...]
nccl_manager.cc 352 ncclSum /* unused */);
359 ncclSum /* unused */);
365 ncclSum /* unused */);
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
nccl_all_reduce_thunk.cc 295 /*op=*/ncclSum,
  /external/tensorflow/tensorflow/core/kernels/
collective_nccl_reducer.cc 63 *reduction_op = ncclSum;
nccl_ops.cc 71 reduction_op_ = ncclSum;

Completed in 859 milliseconds