Searched refs:task_address (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | server_lib.py | 356 ret[job] = {i: self.task_address(job, i) for i in task_indices}
409 def task_address(self, job_name, task_index): member in ClusterSpec
481 for i, task_address in sorted(tasks.items()):
483 task_address = compat.as_bytes(task_address)
486 "Task address %r must be bytes or unicode" % task_address)
487 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/distribute/cluster_resolver/ |
| D | cluster_resolver.py | 225 master = self.cluster_spec().task_address(task_type, task_id)
418 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 | tpu_cluster_resolver.py | 361 master = cluster_spec.task_address(task_type, task_id)
364 master = cluster_spec.task_address(self.task_type, self.task_id)
|
| D | tfconfig_cluster_resolver.py | 171 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/v1/ |
| D | tensorflow.train.-cluster-spec.pbtxt | 30 name: "task_address"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.train.-cluster-spec.pbtxt | 30 name: "task_address"
|
| /external/tensorflow/tensorflow/contrib/tpu/profiler/pip_package/cloud_tpu_profiler/ |
| D | main.py | 79 cluster_spec.task_address(JOB_NAME, i).split(':')[0] for i in task_indices
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_coordinator.py | 411 target = cluster_spec.task_address(task_type, task_id)
|