Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.cc292 CHECK_EQ(gp.device_names.size(), gp.task_names.size()); in BuildDevRecs()
295 TaskDeviceMap& tdm = gdm[gp.task_names[i]]; in BuildDevRecs()
297 dr->task = gp.task_names[i]; in BuildDevRecs()
434 for (int i = 0; i < gp.task_names.size(); ++i) { in EstablishGlobalRank()
435 const string& task_name = gp.task_names[i]; in EstablishGlobalRank()
454 const string* last_task_name = &gp->task_names[0]; in SetDevPerTask()
456 for (const string& task_name : gp->task_names) { in SetDevPerTask()
487 CHECK_EQ(gp->group_size, gp->task_names.size()); in SortDevicesAndTasks()
503 new_tasks.push_back(gp->task_names[pi]); in SortDevicesAndTasks()
506 gp->task_names = std::move(new_tasks); in SortDevicesAndTasks()
[all …]
Dhierarchical_tree_broadcaster.cc88 << absl::StrJoin(col_params->group.task_names, ", "); in InitializeCollectiveParams()
90 const string* prior_task_name = &col_params->group.task_names[0]; in InitializeCollectiveParams()
93 if (col_params->group.task_names[di] != *prior_task_name) { in InitializeCollectiveParams()
96 prior_task_name = &col_params->group.task_names[di]; in InitializeCollectiveParams()
423 col_params_->group.task_names[dst_idx], send_buf_key, col_ctx_->device, in DispatchSend()
443 col_params_->group.task_names[src_idx], in DispatchRecv()
Dring_alg.cc178 << absl::StrJoin(col_params->group.task_names, ", "); in InitializeCollectiveParams()
180 const string* prior_task_name = &col_params->group.task_names[0]; in InitializeCollectiveParams()
183 if (col_params->group.task_names[di] != *prior_task_name) { in InitializeCollectiveParams()
186 prior_task_name = &col_params->group.task_names[di]; in InitializeCollectiveParams()
394 col_params_->group.task_names[send_to_dev_idx], send_buf_key, in DispatchSend()
414 col_params_->group.task_names[rf->recv_dev_idx], in DispatchRecv()
Dpermuter.cc90 col_params_->group.task_names[target_rank], send_buf_key, in DispatchSend()
106 col_params_->group.task_names[src_rank], in DispatchRecv()
Dpermuter_test.cc197 col_params_->group.task_names.push_back(task_name); in Init()
340 col_params_->group.task_names = parent_->col_params_->group.task_names; in DeviceInstance()
Dhierarchical_tree_broadcaster_test.cc341 col_params_->group.task_names.push_back(task_name); in Init()
551 cp->group.task_names.push_back(task_name); in PrepColParamsForSubdivPermsTest()
804 cp->group.task_names.push_back(task_name); in TEST_F()
Dring_gatherer_test.cc232 col_params_->group.task_names.push_back(task_name); in Init()
559 cp->group.task_names.push_back(task_name); in SetUpCollectiveParams()
Dring_reducer_test.cc256 col_params_->group.task_names.push_back(task_name); in Init()
591 cp->group.task_names.push_back(task_name); in SetUpCollectiveParams()
Dbase_collective_executor.cc460 col_params.group.task_names[col_params.default_rank]; in UnblockDependencies()
Dcollective_param_resolver_local_test.cc100 group.task_names.push_back("/job:localhost/replica:0/task:0"); in TEST_F()
/external/tensorflow/tensorflow/core/nccl/
Dcollective_communicator.cc75 col_params->group.task_names[col_params->default_rank]); in Enqueue()
112 << col_params->group.task_names[col_params->default_rank] in Enqueue()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed_test.cc265 EXPECT_EQ(cp_[device_name]->group.task_names[idx], task_name); in ValidateCollectiveParams()
273 EXPECT_EQ(cp_[dev0]->group.task_names[i], in ValidateCollectiveParams()
274 cp_[device_name]->group.task_names[i]); in ValidateCollectiveParams()
/external/perfetto/protos/perfetto/metrics/android/
DBUILD.gn46 "task_names.proto",
/external/tensorflow/tensorflow/core/framework/
Dcollective.cc65 for (const auto& n : task_names) { in ToString()
Dcollective.h69 std::vector<string> task_names; member
/external/perfetto/protos/perfetto/metrics/
Dmetrics.proto43 import "protos/perfetto/metrics/android/task_names.proto";
Dperfetto_merged_metrics.proto935 // Begin of protos/perfetto/metrics/android/task_names.proto
957 // End of protos/perfetto/metrics/android/task_names.proto
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_test.cc145 col_params_->group.task_names.push_back(task_name); in Init()
/external/perfetto/
DBUILD2452 "protos/perfetto/metrics/android/task_names.proto",
DAndroid.bp3672 "protos/perfetto/metrics/android/task_names.proto",
3721 "protos/perfetto/metrics/android/task_names.proto",