Searched refs:task_indices (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | server_lib.py | 81 task_indices = cluster_spec.task_indices(job_name)
82 if len(task_indices) == 1:
83 task_index = task_indices[0]
346 task_indices = self.task_indices(job)
347 if len(task_indices) == 0:
350 if max(task_indices) + 1 == len(task_indices):
356 ret[job] = {i: self.task_address(job, i) for i in task_indices}
390 def task_indices(self, job_name): member in ClusterSpec
|
| D | server_lib_test.py | 515 self.assertEqual([0, 1], cluster_spec.task_indices("ps"))
516 self.assertEqual([0, 1, 2], cluster_spec.task_indices("worker"))
517 self.assertEqual([0, 3], cluster_spec.task_indices("sparse"))
519 cluster_spec.task_indices("unknown")
|
| /external/tensorflow/tensorflow/contrib/tpu/profiler/pip_package/cloud_tpu_profiler/ |
| D | main.py | 77 task_indices = cluster_spec.task_indices(JOB_NAME)
79 cluster_spec.task_address(JOB_NAME, i).split(':')[0] for i in task_indices
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.-cluster-spec.pbtxt | 34 name: "task_indices"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.train.-cluster-spec.pbtxt | 34 name: "task_indices"
|