| /external/tensorflow/tensorflow/core/kernels/ |
| collective_nccl_reducer.cc | 54 if (col_params->default_rank == 0 && col_params->group.num_tasks > 1) {
137 col_params_->instance.task_names[col_params_->default_rank]);
147 col_params_->default_rank, std::move(done_callback));
150 << col_params_->instance.task_names[col_params_->default_rank]
|
| collective_nccl_reducer_test.cc | 223 col_params_.default_rank = rank;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| hierarchical_tree_broadcaster_test.cc | 333 int default_rank = wi * num_devices_per_worker + di; local
335 default_rank, col_params_.instance.device_names[default_rank],
580 col_params_.default_rank = rank;
733 cp.default_rank = 0;
738 cp.default_rank = 2;
743 cp.default_rank = 0;
753 cp.default_rank = 0;
764 cp.default_rank = 0;
775 cp.default_rank = 9
[all...] |
| collective_param_resolver_local.cc | 394 cp->default_rank = i;
414 ir->shared.default_rank = -1;
722 CHECK_GE(cp->default_rank, 0);
723 if (!ir->known[cp->default_rank]) {
724 ir->known[cp->default_rank] = true;
732 cp->default_rank);
734 ir->source_rank = cp->default_rank;
737 if (init_nccl && cp->default_rank == 0) {
745 cp->instance.device_names[cp->default_rank]);
769 cp->instance.device_names[cp->default_rank],
[all...] |
| ring_alg.cc | 166 col_params->instance.device_names[col_params->default_rank];
230 DCHECK_EQ(permuted_di, col_params->default_rank);
384 VLOG(3) << "DispatchSend rank=" << col_params_->default_rank << " send key "
403 VLOG(3) << "DispatchRecv rank=" << col_params_->default_rank << " recv key "
|
| collective_param_resolver_local_test.cc | 198 EXPECT_EQ(cps[i].default_rank, i);
250 EXPECT_EQ(cps[i].default_rank, i);
|
| ring_reducer_test.cc | 426 col_params_.default_rank = my_device_id;
587 cp.default_rank = 0;
604 cp.default_rank = 3;
621 cp.default_rank = 0;
651 cp.default_rank = 0;
|
| ring_gatherer.cc | 86 << " default_rank " << col_params_->default_rank << "\n"
|
| ring_reducer.cc | 78 << " default_rank " << col_params_->default_rank << "\n"
|
| base_collective_executor.cc | 337 col_params.instance.task_names[col_params.default_rank];
|
| ring_gatherer_test.cc | 402 col_params_.default_rank = my_device_id;
560 cp.default_rank = 0;
571 cp.default_rank = 3;
|
| hierarchical_tree_broadcaster.cc | 82 col_params->instance.device_names[col_params->default_rank];
|
| /external/tensorflow/tensorflow/core/framework/ |
| collective.cc | 141 strings::StrAppend(&v, " default_rank=", default_rank,
173 device_name(col_params.instance.device_names[col_params.default_rank]) {}
|
| collective.h | 117 int default_rank = -1; // index of this op within device_names member in struct:tensorflow::CollectiveParams
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| collective_param_resolver_distributed_test.cc | 257 // names, with unique default_rank in the expected order.
265 EXPECT_EQ(cp_[idx].default_rank, idx);
|
