Searched refs:task_address (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | server_lib.py | 365 ret[job] = {i: self.task_address(job, i) for i in task_indices}
418 def task_address(self, job_name, task_index): member in ClusterSpec
490 for i, task_address in sorted(tasks.items()):
492 task_address = compat.as_bytes(task_address)
495 task_address)
496 job_def.tasks[i] = task_address
|
| D | server_lib_test.py | 560 self.assertEqual("ps0:2222", cluster_spec.task_address("ps", 0))
561 self.assertEqual("sparse0:2222", cluster_spec.task_address("sparse", 0))
563 cluster_spec.task_address("unknown", 0)
565 cluster_spec.task_address("sparse", 2)
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | remote.py | 193 task_address = cluster_spec.task_address(job_name, task_id)
194 if master in task_address or task_address in master:
|
| /external/ruy/ruy/ |
| D | thread_pool.cc | 182 auto task_address = reinterpret_cast<std::uintptr_t>(tasks) + i * stride; in ExecuteImpl() local
183 threads_[i - 1]->StartWork(reinterpret_cast<Task*>(task_address)); in ExecuteImpl()
|
| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/tpu/ |
| D | tpu_cluster_resolver.py | 260 master = cluster_spec.task_address(task_type, task_id)
263 master = cluster_spec.task_address(self.task_type, self.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/distribute/cluster_resolver/ |
| D | cluster_resolver.py | 366 master = self.cluster_spec().task_address(task_type, task_id)
590 master = self.cluster_spec().task_address(task_type, task_id)
|
| D | kubernetes_cluster_resolver.py | 144 self.cluster_spec().task_address(task_type, task_id),
|
| D | tfconfig_cluster_resolver.py | 203 return format_master_url(cluster_spec.task_address(task_type, task_id),
|
| D | gce_cluster_resolver.py | 179 master = self.cluster_spec().task_address(task_type, task_id)
|
| D | sagemaker_cluster_resolver.py | 210 cluster_spec.task_address(task_type, task_id), rpc_layer)
|
| D | slurm_cluster_resolver.py | 390 self.cluster_spec().task_address(task_type, task_id),
|
| /external/tensorflow/tensorflow/python/tpu/profiler/ |
| D | capture_tpu_profile.py | 109 cluster_spec.task_address(worker_job_name, i).replace(':8470', ':8466')
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_coordinator.py | 414 target = cluster_spec.task_address(task_type, task_id)
|