Home
last modified time | relevance | path

Searched refs:default_rank (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer.cc54 if (col_params->default_rank == 0 && col_params->group.num_tasks > 1) { in InitializeInstanceBeforeGroupDiscovery()
137 col_params_->instance.task_names[col_params_->default_rank]); in Run()
147 col_params_->default_rank, std::move(done_callback)); in Run()
150 << col_params_->instance.task_names[col_params_->default_rank] in Run()
Dcollective_nccl_reducer_test.cc223 col_params_.default_rank = rank; in DeviceInstance()
/external/tensorflow/tensorflow/core/common_runtime/
Dhierarchical_tree_broadcaster_test.cc333 int default_rank = wi * num_devices_per_worker + di; in Init() local
335 default_rank, col_params_.instance.device_names[default_rank], in Init()
580 col_params_.default_rank = rank; in DeviceInstance()
733 cp.default_rank = 0; in TEST_F()
738 cp.default_rank = 2; in TEST_F()
743 cp.default_rank = 0; in TEST_F()
753 cp.default_rank = 0; in TEST_F()
764 cp.default_rank = 0; in TEST_F()
775 cp.default_rank = 9; in TEST_F()
806 cp.default_rank = 0; in TEST_F()
[all …]
Dcollective_param_resolver_local.cc394 cp->default_rank = i; in SetDefaultRank()
414 ir->shared.default_rank = -1; in InitInstanceSharedParams()
722 CHECK_GE(cp->default_rank, 0); in WaitForGroup()
723 if (!ir->known[cp->default_rank]) { in WaitForGroup()
724 ir->known[cp->default_rank] = true; in WaitForGroup()
732 cp->default_rank); in WaitForGroup()
734 ir->source_rank = cp->default_rank; in WaitForGroup()
737 if (init_nccl && cp->default_rank == 0) { in WaitForGroup()
745 cp->instance.device_names[cp->default_rank]); in WaitForGroup()
769 cp->instance.device_names[cp->default_rank], in WaitForGroup()
Dring_alg.cc166 col_params->instance.device_names[col_params->default_rank]; in InitializeCollectiveParams()
230 DCHECK_EQ(permuted_di, col_params->default_rank); in InitializeCollectiveParams()
384 VLOG(3) << "DispatchSend rank=" << col_params_->default_rank << " send key " in DispatchSend()
403 VLOG(3) << "DispatchRecv rank=" << col_params_->default_rank << " recv key " in DispatchRecv()
Dring_reducer_test.cc426 col_params_.default_rank = my_device_id; in DeviceInstance()
587 cp.default_rank = 0; in TEST_F()
604 cp.default_rank = 3; in TEST_F()
621 cp.default_rank = 0; in TEST_F()
651 cp.default_rank = 0; in TEST_F()
Dcollective_param_resolver_local_test.cc198 EXPECT_EQ(cps[i].default_rank, i); in TEST_F()
250 EXPECT_EQ(cps[i].default_rank, i); in TEST_F()
Dring_gatherer_test.cc402 col_params_.default_rank = my_device_id; in DeviceInstance()
560 cp.default_rank = 0; in TEST_F()
571 cp.default_rank = 3; in TEST_F()
Dring_gatherer.cc86 << " default_rank " << col_params_->default_rank << "\n" in Run()
Dring_reducer.cc78 << " default_rank " << col_params_->default_rank << "\n" in Run()
Dbase_collective_executor.cc337 col_params.instance.task_names[col_params.default_rank]; in Launched()
Dhierarchical_tree_broadcaster.cc82 col_params->instance.device_names[col_params->default_rank]; in InitializeCollectiveParams()
/external/tensorflow/tensorflow/core/framework/
Dcollective.cc141 strings::StrAppend(&v, " default_rank=", default_rank, in ToString()
173 device_name(col_params.instance.device_names[col_params.default_rank]) {} in CollectiveContext()
Dcollective.h117 int default_rank = -1; // index of this op within device_names member
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed_test.cc265 EXPECT_EQ(cp_[idx].default_rank, idx); in ValidateCollectiveParams()