Searched refs:_TaskType (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_coordinator.py | 40 class _TaskType(object): class
99 return len(cluster_spec.as_dict().get(_TaskType.WORKER, [])) + len(
100 cluster_spec.as_dict().get(_TaskType.CHIEF, []))
172 if not self._cluster_spec or self._task_type == _TaskType.EVALUATOR:
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
[all …]
|
| D | estimator_training.py | 32 CHIEF = dc._TaskType.CHIEF
33 EVALUATOR = dc._TaskType.EVALUATOR
34 PS = dc._TaskType.PS
35 WORKER = dc._TaskType.WORKER
|
| D | distribute_coordinator_test.py | 56 CHIEF = distribute_coordinator._TaskType.CHIEF
57 WORKER = distribute_coordinator._TaskType.WORKER
58 PS = distribute_coordinator._TaskType.PS
59 EVALUATOR = distribute_coordinator._TaskType.EVALUATOR
|
| /external/llvm-project/pstl/include/pstl/internal/ |
| D | parallel_backend_tbb.h | 1273 _TaskType; in __parallel_merge()
1274 __root_task<_TaskType> __root{__xs, __xe, __ys, __ye, __zs, __comp, __leaf_merge}; in __parallel_merge()
|