Home
last modified time | relevance | path

Searched refs:dev_per_task (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dhierarchical_tree_broadcaster.cc63 int device_rank, const std::vector<int>& dev_per_task) { in GetDeviceTask() argument
64 int num_tasks = static_cast<int>(dev_per_task.size()); in GetDeviceTask()
68 task_hi = task_lo + dev_per_task[ti]; in GetDeviceTask()
89 std::vector<int> dev_per_task; in InitializeCollectiveParams() local
94 dev_per_task.push_back(dev_count); in InitializeCollectiveParams()
101 dev_per_task.push_back(dev_count); in InitializeCollectiveParams()
102 CHECK_EQ(col_params->group.num_tasks, dev_per_task.size()); in InitializeCollectiveParams()
106 for (int dpt : dev_per_task) strings::StrAppend(&dpt_buf, dpt, ";"); in InitializeCollectiveParams()
117 for (int num_dev : dev_per_task) total_num_devices += num_dev; in InitializeCollectiveParams()
132 int source_task = GetDeviceTask(col_params->source_rank, dev_per_task); in InitializeCollectiveParams()
[all …]
Dring_alg.cc179 std::vector<int> dev_per_task; in InitializeCollectiveParams() local
184 dev_per_task.push_back(dev_count); in InitializeCollectiveParams()
191 dev_per_task.push_back(dev_count); in InitializeCollectiveParams()
192 DCHECK_EQ(col_params->group.num_tasks, dev_per_task.size()); in InitializeCollectiveParams()
222 for (int di = 0; di < dev_per_task[ti]; ++di) { in InitializeCollectiveParams()
223 int di_offset = (di + offset) % dev_per_task[ti]; in InitializeCollectiveParams()
225 reverse ? (dev_per_task[ti] - (di_offset + 1)) : di_offset; in InitializeCollectiveParams()
235 prior_dev_count += dev_per_task[ti]; in InitializeCollectiveParams()
Dhierarchical_tree_broadcaster.h68 int GetDeviceTask(int device_rank, const std::vector<int>& dev_per_task);
Dcollective_param_resolver_local.cc468 int dev_per_task = -1; in SetDevPerTask() local
470 if (dev_per_task == -1) { in SetDevPerTask()
471 dev_per_task = task_dev.second; in SetDevPerTask()
472 } else if (dev_per_task != task_dev.second) { in SetDevPerTask()
Dhierarchical_tree_broadcaster_test.cc799 std::vector<int> dev_per_task = {4, 4, 6, 8}; in TEST_F() local
802 for (int di = 0; di < dev_per_task[ti]; di++) { in TEST_F()