Searched refs:task_indices (Results 1 – 6 of 6) sorted by relevance
80 task_indices = cluster_spec.task_indices(job_name)81 if len(task_indices) == 1:82 task_index = task_indices[0]355 task_indices = self.task_indices(job)356 if len(task_indices) == 0:359 if max(task_indices) + 1 == len(task_indices):365 ret[job] = {i: self.task_address(job, i) for i in task_indices}399 def task_indices(self, job_name): member in ClusterSpec
567 self.assertEqual([0, 1], cluster_spec.task_indices("ps"))568 self.assertEqual([0, 1, 2], cluster_spec.task_indices("worker"))569 self.assertEqual([0, 3], cluster_spec.task_indices("sparse"))571 cluster_spec.task_indices("unknown")
107 task_indices = cluster_spec.task_indices(worker_job_name)110 for i in task_indices
34 name: "task_indices"
192 for task_id in cluster_spec.task_indices(job_name):