| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_param_resolver_local.cc | 226 CHECK_EQ(ip.device_names.size(), ip.task_names.size()); in BuildDevRecs()
229 TaskDeviceMap& tdm = gdm[ip.task_names[i]]; in BuildDevRecs()
231 dr->task = ip.task_names[i]; in BuildDevRecs()
370 for (int i = 0; i < cp->instance.task_names.size(); ++i) { in EstablishGlobalRank()
371 const string& task_name = cp->instance.task_names[i]; in EstablishGlobalRank()
390 const string* last_task_name = &cp->instance.task_names[0]; in SetDevPerTask()
392 for (const string& task_name : cp->instance.task_names) { in SetDevPerTask()
423 CHECK_EQ(cp->group.group_size, cp->instance.task_names.size()); in SortDevicesAndTasks()
439 new_tasks.push_back(cp->instance.task_names[pi]); in SortDevicesAndTasks()
442 cp->instance.task_names = std::move(new_tasks); in SortDevicesAndTasks()
[all …]
|
| D | ring_alg.cc | 177 << absl::StrJoin(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()
|
| D | hierarchical_tree_broadcaster.cc | 89 << absl::StrJoin(col_params->instance.task_names, ", "); in InitializeCollectiveParams()
91 const string* prior_task_name = &col_params->instance.task_names[0]; in InitializeCollectiveParams()
94 if (col_params->instance.task_names[di] != *prior_task_name) { in InitializeCollectiveParams()
97 prior_task_name = &col_params->instance.task_names[di]; in InitializeCollectiveParams()
421 col_params_->instance.task_names[dst_idx], in DispatchSend()
441 col_params_->instance.task_names[src_idx], in DispatchRecv()
|
| D | hierarchical_tree_broadcaster_test.cc | 333 col_params_.instance.task_names.push_back(task_name); in Init()
553 cp->instance.task_names.push_back(task_name); in PrepColParamsForSubdivPermsTest()
578 col_params_.instance.task_names = in DeviceInstance()
579 parent_->col_params_.instance.task_names; in DeviceInstance()
804 cp.instance.task_names.push_back(task_name); in TEST_F()
|
| D | ring_gatherer_test.cc | 227 col_params_.instance.task_names.push_back(task_name); in Init()
554 cp.instance.task_names.push_back(task_name); in SetUpCollectiveParams()
|
| D | base_collective_executor.cc | 359 col_params.instance.task_names[col_params.default_rank]; in UnblockDependencies()
|
| D | ring_reducer_test.cc | 250 col_params_.instance.task_names.push_back(task_name); in Init()
582 cp.instance.task_names.push_back(task_name); in SetUpCollectiveParams()
|
| D | collective_param_resolver_local_test.cc | 95 cp.instance.task_names.push_back("/job:localhost/replica:0/task:0"); in TEST_F()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 72 task_names.assign(other.task_names.begin(), other.task_names.end()); in operator =()
101 for (const auto& n : task_names) { in ToString()
|
| D | collective.h | 99 std::vector<string> task_names; member
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_gatherer.cc | 31 col_params_->instance.task_names[col_params_->default_rank]); in Run()
39 << col_params_->instance.task_names[col_params_->default_rank] in Run()
|
| D | collective_nccl_broadcaster.cc | 31 col_params_->instance.task_names[col_params_->default_rank]); in Run()
40 << col_params_->instance.task_names[col_params_->default_rank] in Run()
|
| D | collective_nccl_reducer.cc | 133 col_params_->instance.task_names[col_params_->default_rank]); in Run()
141 << col_params_->instance.task_names[col_params_->default_rank] in Run()
|
| D | collective_nccl_test.cc | 138 col_params_.instance.task_names.push_back(task_name); in Init()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed_test.cc | 271 EXPECT_EQ(cp_[idx].instance.task_names[idx], task_name); in ValidateCollectiveParams()
278 EXPECT_EQ(cp_[0].instance.task_names[i], in ValidateCollectiveParams()
279 cp_[idx].instance.task_names[i]); in ValidateCollectiveParams()
|
| /external/perfetto/protos/perfetto/metrics/android/ |
| D | BUILD.gn | 35 "task_names.proto",
|
| /external/perfetto/protos/perfetto/metrics/ |
| D | metrics.proto | 38 import "protos/perfetto/metrics/android/task_names.proto";
|
| D | perfetto_merged_metrics.proto | 645 // Begin of protos/perfetto/metrics/android/task_names.proto
667 // End of protos/perfetto/metrics/android/task_names.proto
|
| /external/perfetto/ |
| D | Android.bp | 3116 "protos/perfetto/metrics/android/task_names.proto",
3143 "external/perfetto/protos/perfetto/metrics/android/task_names.pbzero.cc",
3170 "protos/perfetto/metrics/android/task_names.proto",
3197 "external/perfetto/protos/perfetto/metrics/android/task_names.pbzero.h",
|
| D | BUILD | 2026 "protos/perfetto/metrics/android/task_names.proto",
|