Searched refs:task_address (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | server_lib.py | 360 ret[job] = {i: self.task_address(job, i) for i in task_indices}
413 def task_address(self, job_name, task_index): member in ClusterSpec
485 for i, task_address in sorted(tasks.items()):
487 task_address = compat.as_bytes(task_address)
490 task_address)
491 job_def.tasks[i] = task_address
|
| D | server_lib_test.py | 508 self.assertEqual("ps0:2222", cluster_spec.task_address("ps", 0))
509 self.assertEqual("sparse0:2222", cluster_spec.task_address("sparse", 0))
511 cluster_spec.task_address("unknown", 0)
513 cluster_spec.task_address("sparse", 2)
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | remote.py | 194 task_address = cluster_spec.task_address(job_name, task_id)
195 if master in task_address or task_address in master:
|
| /external/tensorflow/tensorflow/lite/experimental/ruy/ |
| D | thread_pool.cc | 169 auto task_address = reinterpret_cast<std::uintptr_t>(tasks) + i * stride; in ExecuteImpl() local
170 threads_[i - 1]->StartWork(reinterpret_cast<Task*>(task_address)); in ExecuteImpl()
|
| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| D | tpu_cluster_resolver.py | 204 master = cluster_spec.task_address(task_type, task_id)
207 master = cluster_spec.task_address(self.task_type, self.task_id)
|
| D | cluster_resolver.py | 219 master = self.cluster_spec().task_address(task_type, task_id)
412 master = self.cluster_spec().task_address(task_type, task_id)
|
| D | kubernetes_cluster_resolver.py | 117 self.cluster_spec().task_address(task_type, task_id),
|
| D | tfconfig_cluster_resolver.py | 177 return format_master_url(cluster_spec.task_address(task_type, task_id),
|
| D | gce_cluster_resolver.py | 156 master = self.cluster_spec().task_address(task_type, task_id)
|
| D | slurm_cluster_resolver.py | 215 self.cluster_spec().task_address(task_type, task_id),
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.train.-cluster-spec.pbtxt | 30 name: "task_address"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.-cluster-spec.pbtxt | 30 name: "task_address"
|
| /external/tensorflow/tensorflow/python/tpu/profiler/ |
| D | capture_tpu_profile.py | 108 cluster_spec.task_address(worker_job_name, i).split(':')[0]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_coordinator.py | 414 target = cluster_spec.task_address(task_type, task_id)
|