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.cc62 int device_rank, const std::vector<int>& dev_per_task) { in GetDeviceTask() argument
63 int num_tasks = static_cast<int>(dev_per_task.size()); in GetDeviceTask()
67 task_hi = task_lo + dev_per_task[ti]; in GetDeviceTask()
88 std::vector<int> dev_per_task; in InitializeCollectiveParams() local
93 dev_per_task.push_back(dev_count); in InitializeCollectiveParams()
100 dev_per_task.push_back(dev_count); in InitializeCollectiveParams()
101 CHECK_EQ(col_params->group.num_tasks, dev_per_task.size()); in InitializeCollectiveParams()
105 for (int dpt : dev_per_task) strings::StrAppend(&dpt_buf, dpt, ";"); in InitializeCollectiveParams()
116 for (int num_dev : dev_per_task) total_num_devices += num_dev; in InitializeCollectiveParams()
131 int source_task = GetDeviceTask(col_params->source_rank, dev_per_task); in InitializeCollectiveParams()
[all …]
Dring_alg.cc178 std::vector<int> dev_per_task; in InitializeCollectiveParams() local
183 dev_per_task.push_back(dev_count); in InitializeCollectiveParams()
190 dev_per_task.push_back(dev_count); in InitializeCollectiveParams()
191 DCHECK_EQ(col_params->group.num_tasks, dev_per_task.size()); in InitializeCollectiveParams()
221 for (int di = 0; di < dev_per_task[ti]; ++di) { in InitializeCollectiveParams()
222 int di_offset = (di + offset) % dev_per_task[ti]; in InitializeCollectiveParams()
224 reverse ? (dev_per_task[ti] - (di_offset + 1)) : di_offset; in InitializeCollectiveParams()
234 prior_dev_count += dev_per_task[ti]; in InitializeCollectiveParams()
Dhierarchical_tree_broadcaster.h66 int GetDeviceTask(int device_rank, const std::vector<int>& dev_per_task);
Dcollective_param_resolver_local.cc337 int dev_per_task = -1; in SetDevPerTask() local
339 if (dev_per_task == -1) { in SetDevPerTask()
340 dev_per_task = task_dev.second; in SetDevPerTask()
341 } else if (dev_per_task != task_dev.second) { in SetDevPerTask()
Dhierarchical_tree_broadcaster_test.cc793 std::vector<int> dev_per_task = {4, 4, 6, 8}; in TEST_F() local
796 for (int di = 0; di < dev_per_task[ti]; di++) { in TEST_F()