Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_gatherer.cc31 col_params_->instance.task_names[col_params_->default_rank]); in Run()
36 col_ctx_->output, col_params_->default_rank, std::move(done)); in Run()
39 << col_params_->instance.task_names[col_params_->default_rank] in Run()
42 << col_params_->default_rank << " device " << col_ctx_->device_name in Run()
Dcollective_nccl_broadcaster.cc31 col_params_->instance.task_names[col_params_->default_rank]); in Run()
36 col_ctx_->output, col_params_->default_rank, std::move(done)); in Run()
40 << col_params_->instance.task_names[col_params_->default_rank] in Run()
42 << num_global_devices << " rank " << col_params_->default_rank in Run()
Dcollective_nccl_reducer.cc133 col_params_->instance.task_names[col_params_->default_rank]); in Run()
138 col_ctx_->output, col_params_->default_rank, std::move(done_callback)); in Run()
141 << col_params_->instance.task_names[col_params_->default_rank] in Run()
Dcollective_nccl_test.cc253 col_params_.default_rank = rank; in DeviceInstance()
347 << col_params_.default_rank; in RunBroadcast()
378 << col_params_.default_rank; in RunGather()
493 di->col_params_.is_source = di->col_params_.default_rank == source_rank_; in InitDevice()
/external/tensorflow/tensorflow/core/common_runtime/
Dhierarchical_tree_broadcaster_test.cc339 int default_rank = wi * num_devices_per_worker + di; in Init() local
341 default_rank, col_params_.instance.device_names[default_rank], in Init()
588 col_params_.default_rank = rank; in DeviceInstance()
739 cp.default_rank = 0; in TEST_F()
744 cp.default_rank = 2; in TEST_F()
749 cp.default_rank = 0; in TEST_F()
759 cp.default_rank = 0; in TEST_F()
770 cp.default_rank = 0; in TEST_F()
781 cp.default_rank = 9; in TEST_F()
812 cp.default_rank = 0; in TEST_F()
[all …]
Dcollective_param_resolver_local.cc461 cp->default_rank = i; in SetDefaultRank()
481 ir->shared.default_rank = -1; in InitInstanceSharedParams()
796 CHECK_GE(cp->default_rank, 0); in WaitForGroup()
797 if (!ir->known[cp->default_rank]) { in WaitForGroup()
798 ir->known[cp->default_rank] = true; in WaitForGroup()
806 cp->default_rank); in WaitForGroup()
808 ir->source_rank = 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.cc434 col_params_.default_rank = my_device_id; in DeviceInstance()
593 cp.default_rank = 0; in TEST_F()
610 cp.default_rank = 3; in TEST_F()
627 cp.default_rank = 0; in TEST_F()
657 cp.default_rank = 0; in TEST_F()
Dcollective_param_resolver_local_test.cc199 EXPECT_EQ(cps[i].default_rank, i); in TEST_F()
251 EXPECT_EQ(cps[i].default_rank, i); in TEST_F()
Dring_gatherer_test.cc409 col_params_.default_rank = my_device_id; in DeviceInstance()
565 cp.default_rank = 0; in TEST_F()
576 cp.default_rank = 3; in TEST_F()
Dring_gatherer.cc89 << " default_rank " << col_params_->default_rank << "\n" in Run()
Dring_reducer.cc85 << " default_rank " << col_params_->default_rank << "\n" in Run()
Dbase_collective_executor.cc359 col_params.instance.task_names[col_params.default_rank]; in UnblockDependencies()
Dhierarchical_tree_broadcaster.cc84 col_params->instance.device_names[col_params->default_rank]; in InitializeCollectiveParams()
/external/tensorflow/tensorflow/core/framework/
Dcollective.cc146 strings::StrAppend(&v, " default_rank=", default_rank, in ToString()
178 device_name(col_params.instance.device_names[col_params.default_rank]) {} in CollectiveContext()
Dcollective.h126 int default_rank = -1; // index of this op within device_names member
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed_test.cc268 EXPECT_EQ(cp_[idx].default_rank, idx); in ValidateCollectiveParams()