Home
last modified time | relevance | path

Searched refs:_TaskType (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Ddistribute_coordinator.py40 class _TaskType(object): class
99 return len(cluster_spec.as_dict().get(_TaskType.WORKER, [])) + len(
100 cluster_spec.as_dict().get(_TaskType.CHIEF, []))
178 if _TaskType.CHIEF in self._cluster_spec.jobs:
179 task_type = _TaskType.CHIEF
182 assert _TaskType.WORKER in self._cluster_spec.jobs
183 task_type = _TaskType.WORKER
197 self._task_type in [_TaskType.CHIEF, _TaskType.EVALUATOR, None]):
202 if (_TaskType.CHIEF not in self._cluster_spec.jobs and
203 self._task_type == _TaskType.WORKER and self._task_id == 0):
[all …]
Destimator_training.py32 CHIEF = dc._TaskType.CHIEF
33 EVALUATOR = dc._TaskType.EVALUATOR
34 PS = dc._TaskType.PS
35 WORKER = dc._TaskType.WORKER
Ddistribute_coordinator_test.py53 CHIEF = distribute_coordinator._TaskType.CHIEF
54 WORKER = distribute_coordinator._TaskType.WORKER
55 PS = distribute_coordinator._TaskType.PS
56 EVALUATOR = distribute_coordinator._TaskType.EVALUATOR
/external/tensorflow/tensorflow/contrib/distribute/python/
Destimator_training_test.py65 CHIEF = dc._TaskType.CHIEF
66 EVALUATOR = dc._TaskType.EVALUATOR
67 WORKER = dc._TaskType.WORKER
68 PS = dc._TaskType.PS