OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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