Home
last modified time | relevance | path

Searched refs:task_names (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.cc162 CHECK_EQ(ip.device_names.size(), ip.task_names.size()); in BuildDevRecs()
165 TaskDeviceMap& tdm = gdm[ip.task_names[i]]; in BuildDevRecs()
167 dr->task = ip.task_names[i]; in BuildDevRecs()
303 for (int i = 0; i < cp->instance.task_names.size(); ++i) { in EstablishGlobalRank()
304 const string& task_name = cp->instance.task_names[i]; in EstablishGlobalRank()
323 const string* last_task_name = &cp->instance.task_names[0]; in SetDevPerTask()
325 for (const string& task_name : cp->instance.task_names) { in SetDevPerTask()
356 CHECK_EQ(cp->group.group_size, cp->instance.task_names.size()); in SortDevicesAndTasks()
372 new_tasks.push_back(cp->instance.task_names[pi]); in SortDevicesAndTasks()
375 cp->instance.task_names = std::move(new_tasks); in SortDevicesAndTasks()
[all …]
Dhierarchical_tree_broadcaster.cc87 << str_util::Join(col_params->instance.task_names, ", "); in InitializeCollectiveParams()
89 const string* prior_task_name = &col_params->instance.task_names[0]; in InitializeCollectiveParams()
92 if (col_params->instance.task_names[di] != *prior_task_name) { in InitializeCollectiveParams()
95 prior_task_name = &col_params->instance.task_names[di]; in InitializeCollectiveParams()
411 col_params_->instance.task_names[dst_idx], in DispatchSend()
431 col_params_->instance.task_names[src_idx], in DispatchRecv()
Dring_alg.cc177 << str_util::Join(col_params->instance.task_names, ", "); in InitializeCollectiveParams()
179 const string* prior_task_name = &col_params->instance.task_names[0]; in InitializeCollectiveParams()
182 if (col_params->instance.task_names[di] != *prior_task_name) { in InitializeCollectiveParams()
185 prior_task_name = &col_params->instance.task_names[di]; in InitializeCollectiveParams()
392 col_params_->instance.task_names[send_to_dev_idx], send_buf_key, in DispatchSend()
411 col_params_->instance.task_names[rf->recv_dev_idx], in DispatchRecv()
Dhierarchical_tree_broadcaster_test.cc327 col_params_.instance.task_names.push_back(task_name); in Init()
545 cp->instance.task_names.push_back(task_name); in PrepColParamsForSubdivPermsTest()
570 col_params_.instance.task_names = in DeviceInstance()
571 parent_->col_params_.instance.task_names; in DeviceInstance()
798 cp.instance.task_names.push_back(task_name); in TEST_F()
Dbase_collective_executor.cc337 col_params.instance.task_names[col_params.default_rank]; in Launched()
Dring_gatherer_test.cc221 col_params_.instance.task_names.push_back(task_name); in Init()
549 cp.instance.task_names.push_back(task_name); in SetUpCollectiveParams()
Dcollective_param_resolver_local_test.cc95 cp.instance.task_names.push_back("/job:localhost/replica:0/task:0"); in TEST_F()
Dring_reducer_test.cc243 col_params_.instance.task_names.push_back(task_name); in Init()
576 cp.instance.task_names.push_back(task_name); in SetUpCollectiveParams()
/external/tensorflow/tensorflow/core/framework/
Dcollective.cc66 task_names.assign(other.task_names.begin(), other.task_names.end()); in operator =()
95 for (const auto& n : task_names) { in ToString()
Dcollective.h88 std::vector<string> task_names; member
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer.cc137 col_params_->instance.task_names[col_params_->default_rank]); in Run()
150 << col_params_->instance.task_names[col_params_->default_rank] in Run()
Dcollective_nccl_reducer_test.cc131 col_params_.instance.task_names.push_back(task_name); in Init()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed_test.cc268 EXPECT_EQ(cp_[idx].instance.task_names[idx], task_name); in ValidateCollectiveParams()
275 EXPECT_EQ(cp_[0].instance.task_names[i], in ValidateCollectiveParams()
276 cp_[idx].instance.task_names[i]); in ValidateCollectiveParams()
Ddevice_resolver_distributed.cc80 GetLocalityAsync(inst_params.device_names[i], inst_params.task_names[i], in GetDeviceLocalitiesRecursive()