Home
last modified time | relevance | path

Searched refs:task_name (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed_test.cc96 string task_name; in GetDeviceLocalityAsync() local
98 if (!DeviceNameUtils::SplitDeviceName(device, &task_name, &dev_part)) { in GetDeviceLocalityAsync()
102 auto it = workers_.find(task_name); in GetDeviceLocalityAsync()
104 done(errors::Internal("failed to find worker ", task_name)); in GetDeviceLocalityAsync()
187 string task_name = strings::StrCat("/job:worker/replica:0/task:", wi); in DefineCollectiveParams() local
189 string device_name = strings::StrCat(task_name, "/device:CPU:", di); in DefineCollectiveParams()
226 string task_name = strings::StrCat("/job:worker/replica:0/task:", wi); in IssueRequest() local
227 string device_name = strings::StrCat(task_name, "/device:CPU:", di); in IssueRequest()
233 CollectiveParamResolverDistributed* cp_res = cp_resolvers_[task_name]; in IssueRequest()
260 string task_name = strings::StrCat("/job:worker/replica:0/task:", wi); in ValidateCollectiveParams() local
[all …]
Drpc_collective_executor_mgr_test.cc38 string task_name = "/job:localhost/replica:0/task:0"; in RpcCollectiveExecutorMgrTest() local
41 task_name); in RpcCollectiveExecutorMgrTest()
46 TF_CHECK_OK(DeviceFactory::AddDevices(options, task_name, &devices)); in RpcCollectiveExecutorMgrTest()
49 device_mgr_.get(), worker_cache, task_name)); in RpcCollectiveExecutorMgrTest()
53 worker_cache, task_name)); in RpcCollectiveExecutorMgrTest()
57 worker_cache, task_name)); in RpcCollectiveExecutorMgrTest()
Ddevice_resolver_distributed_test.cc98 string task_name; in GetDeviceLocalityAsync() local
100 if (!DeviceNameUtils::SplitDeviceName(device, &task_name, &dev_part)) { in GetDeviceLocalityAsync()
104 auto it = workers_.find(task_name); in GetDeviceLocalityAsync()
106 done(errors::Internal("failed to find worker ", task_name)); in GetDeviceLocalityAsync()
187 const string& task_name = it2.first; in TEST_F() local
194 dres->GetLocalityAsync(dev_name, task_name, &locality, in TEST_F()
Drpc_collective_executor_mgr.cc32 WorkerCacheInterface* worker_cache, const string& task_name) in RpcCollectiveExecutorMgr() argument
36 task_name_(task_name) { in RpcCollectiveExecutorMgr()
37 group_leader_ = (task_name == config.experimental().collective_group_leader()) in RpcCollectiveExecutorMgr()
Dcollective_rma_distributed_test.cc136 string task_name; in GetDeviceLocalityAsync() local
138 if (!DeviceNameUtils::SplitDeviceName(device, &task_name, &dev_part)) { in GetDeviceLocalityAsync()
142 auto it = workers_.find(task_name); in GetDeviceLocalityAsync()
144 done(errors::Internal("failed to find worker ", task_name)); in GetDeviceLocalityAsync()
Dcollective_param_resolver_distributed.cc84 const string& task_name) in CollectiveParamResolverDistributed() argument
85 : CollectiveParamResolverLocal(config, dev_mgr, dev_resolver, task_name), in CollectiveParamResolverDistributed()
87 group_leader_(task_name == config.experimental().collective_group_leader() in CollectiveParamResolverDistributed()
90 VLOG(1) << "CompleteParamResolverDistributed ctor task={" << task_name in CollectiveParamResolverDistributed()
236 for (const string& tn : resp.task_name()) { in UpdateGroupCache()
Ddevice_resolver_distributed.cc23 const string& task_name) in DeviceResolverDistributed() argument
24 : dev_mgr_(dev_mgr), worker_cache_(worker_cache), task_name_(task_name) {} in DeviceResolverDistributed()
Ddevice_resolver_distributed.h33 const string& task_name);
Drpc_collective_executor_mgr.h41 WorkerCacheInterface* worker_cache, const string& task_name);
Dcollective_param_resolver_distributed.h32 const string& task_name);
/external/perfetto/src/traced/probes/ftrace/test/data/android_walleye_OPM5.171019.017.A1_4.4.88/events/exception/user_fault/
Dformat9 field:__data_loc char[] task_name; offset:8; size:4; signed:0;
13 print fmt: "task_name:%s addr:%lu, fsr:%u", __get_str(task_name), REC->addr, REC->fsr
/external/perfetto/src/traced/probes/ftrace/test/data/android_seed_N2F62_3.10.49/events/exception/user_fault/
Dformat9 field:__data_loc char[] task_name; offset:8; size:4; signed:0;
13 print fmt: "task_name:%s addr:%lu, fsr:%u", __get_str(task_name), REC->addr, REC->fsr
/external/perfetto/src/traced/probes/ftrace/test/data/android_hammerhead_MRA59G_3.4.0/events/exception/user_fault/
Dformat10 field:__data_loc char[] task_name; offset:12; size:4; signed:0;
14 print fmt: "task_name:%s addr:%lu, fsr:%u", __get_str(task_name), REC->addr, REC->fsr
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.cc42 DeviceResolverInterface* dev_resolver, const string& task_name) in CollectiveParamResolverLocal() argument
46 task_name_(task_name) {} in CollectiveParamResolverLocal()
112 string task_name = strings::StrCat("/job:", parsed_device.job, in CompleteGroupLocal() local
115 gr->task_set.insert(task_name); in CompleteGroupLocal()
116 gr->task_list.push_back(task_name); in CompleteGroupLocal()
304 const string& task_name = cp->instance.task_names[i]; in EstablishGlobalRank() local
305 if (ordered_tasks.find(task_name) != ordered_tasks.end()) { in EstablishGlobalRank()
308 ordered_tasks.insert(task_name); in EstablishGlobalRank()
309 TaskDeviceMap* tdm = &gdm[task_name]; in EstablishGlobalRank()
325 for (const string& task_name : cp->instance.task_names) { in SetDevPerTask() local
[all …]
Dcollective_executor_mgr_test.cc39 string task_name = "/job:localhost/replica:0/task:0"; in CollectiveExecutorMgrTest() local
42 TF_CHECK_OK(DeviceFactory::AddDevices(options, task_name, &devices)); in CollectiveExecutorMgrTest()
48 task_name)); in CollectiveExecutorMgrTest()
Ddevice_resolver_local_test.cc35 string task_name = "/job:localhost/replica:0/task:0"; in DeviceResolverLocalTest() local
39 TF_CHECK_OK(DeviceFactory::AddDevices(options, task_name, &devices)); in DeviceResolverLocalTest()
Dhierarchical_tree_broadcaster_test.cc318 string task_name = strings::StrCat("/job:worker/replica:0/task:", wi); in Init() local
321 dev_name = strings::StrCat(task_name, "/device:GPU:0"); in Init()
323 dev_name = strings::StrCat(task_name, "/device:CPU:", di); in Init()
327 col_params_.instance.task_names.push_back(task_name); in Init()
542 string task_name = strings::StrCat("/job:worker/replica:0/task:", ti); in PrepColParamsForSubdivPermsTest() local
544 string dev_name = strings::StrCat(task_name, "/device:GPU:", di); in PrepColParamsForSubdivPermsTest()
545 cp->instance.task_names.push_back(task_name); in PrepColParamsForSubdivPermsTest()
795 string task_name = strings::StrCat("/job:worker/replica:0/task:", ti); in TEST_F() local
797 string dev_name = strings::StrCat(task_name, "/device:GPU:", di); in TEST_F()
798 cp.instance.task_names.push_back(task_name); in TEST_F()
Dring_gatherer_test.cc214 string task_name = strings::StrCat("/job:worker/replica:0/task:", wi); in Init() local
215 string dev_name = strings::StrCat(task_name, "/cpu:", di); in Init()
218 strings::StrCat(task_name, "/gpu:", di % gpu_devices_.size()); in Init()
221 col_params_.instance.task_names.push_back(task_name); in Init()
547 string task_name = strings::StrCat("/job:worker/replica:0/task:", task_id); in SetUpCollectiveParams() local
548 string device_name = strings::StrCat(task_name, "/device:GPU:", dev_id); in SetUpCollectiveParams()
549 cp.instance.task_names.push_back(task_name); in SetUpCollectiveParams()
Dring_reducer_test.cc236 string task_name = strings::StrCat("/job:worker/replica:0/task:", wi); in Init() local
237 string dev_name = strings::StrCat(task_name, "/cpu:", di); in Init()
240 strings::StrCat(task_name, "/gpu:", di % gpu_devices_.size()); in Init()
243 col_params_.instance.task_names.push_back(task_name); in Init()
574 string task_name = strings::StrCat("/job:worker/replica:0/task:", task_id); in SetUpCollectiveParams() local
575 string device_name = strings::StrCat(task_name, "/device:GPU:", dev_id); in SetUpCollectiveParams()
576 cp.instance.task_names.push_back(task_name); in SetUpCollectiveParams()
Dcollective_param_resolver_local_test.cc37 string task_name = "/job:localhost/replica:0/task:0"; in CollectiveParamResolverLocalTest() local
41 TF_CHECK_OK(DeviceFactory::AddDevices(options, task_name, &devices)); in CollectiveParamResolverLocalTest()
45 drl_.get(), task_name)); in CollectiveParamResolverLocalTest()
Dcollective_param_resolver_local.h43 const string& task_name);
217 void CompleteTaskIsLocal(const string& task_name, CollectiveParams* cp);
Dbase_collective_executor.cc336 const string& task_name = in Launched() local
339 col_params.instance.num_devices_per_task.at(task_name); in Launched()
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_collective_executor_mgr.h39 WorkerCacheInterface* worker_cache, const string& task_name, in GdrCollectiveExecutorMgr() argument
43 task_name), in GdrCollectiveExecutorMgr()
/external/autotest/venv/skylab_staging/
Dswarming.py121 task_name = '%s-%s' % (build, suite_name)
132 'task_name': task_name,
145 '--task-name', task_name,
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer_test.cc126 const string task_name = "/job:worker/replica:0/task:0"; in Init() local
127 col_params_.instance.num_devices_per_task[task_name] = num_ranks; in Init()
131 col_params_.instance.task_names.push_back(task_name); in Init()

12