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.cc64 int device_rank, const std::vector<int>& dev_per_task) { in GetDeviceTask() argument
65 int num_tasks = static_cast<int>(dev_per_task.size()); in GetDeviceTask()
69 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.cc192 std::vector<int> dev_per_task; in InitializeCollectiveParams() local
197 dev_per_task.push_back(dev_count); in InitializeCollectiveParams()
204 dev_per_task.push_back(dev_count); in InitializeCollectiveParams()
205 DCHECK_EQ(col_params->group.num_tasks, dev_per_task.size()); in InitializeCollectiveParams()
235 for (int di = 0; di < dev_per_task[ti]; ++di) { in InitializeCollectiveParams()
236 int di_offset = (di + offset) % dev_per_task[ti]; in InitializeCollectiveParams()
238 reverse ? (dev_per_task[ti] - (di_offset + 1)) : di_offset; in InitializeCollectiveParams()
249 prior_dev_count += dev_per_task[ti]; in InitializeCollectiveParams()
Dhierarchical_tree_broadcaster.h62 int GetDeviceTask(int device_rank, const std::vector<int>& dev_per_task);
Dcollective_param_resolver_local.cc475 int dev_per_task = -1; in SetDevPerTask() local
477 if (dev_per_task == -1) { in SetDevPerTask()
478 dev_per_task = task_dev.second; in SetDevPerTask()
479 } else if (dev_per_task != task_dev.second) { in SetDevPerTask()
Dhierarchical_tree_broadcaster_test.cc358 std::vector<int> dev_per_task = {4, 4, 6, 8}; in TEST_F() local
361 for (int di = 0; di < dev_per_task[ti]; di++) { in TEST_F()