Searched refs:_is_chief (Results 1 – 16 of 16) sorted by relevance
153 self._is_chief = None155 self._is_chief = (self._task_id == 0)159 self._is_chief = (self._task_type == TaskType.MASTER and163 self._is_chief = (self._task_type == TaskType.WORKER and182 return self._is_chief
89 self._is_chief = is_chief99 elif self._is_chief:
135 self._is_chief = is_chief246 if self._is_chief:293 return _ModelAverageOptimizerHook(self, self._is_chief)306 self._is_chief = is_chief311 if self._is_chief:
250 self._is_chief = is_chief256 if self._is_chief:
456 self._is_chief = is_chief462 if self._is_chief:
192 self._is_chief = multi_worker_util.is_chief(cluster_spec, task_type,203 num_ps_replicas, self._is_chief, self._device_map,243 self._is_chief = True538 return self._is_chief542 return self._is_chief546 return self._is_chief
108 self._is_chief = True179 self._is_chief = multi_worker_util.is_chief(cluster_spec, task_type,282 if self._is_chief:463 return self._is_chief467 return self._is_chief
146 self._is_chief_node = self._is_chief()194 def _is_chief(self): member in _WorkerContext
100 config._is_chief = worker_context.is_chief
327 self._is_chief = is_chief343 if self._is_chief:512 return self._is_chief631 assert self._is_chief663 if not self._is_chief:725 if self._is_chief:921 if not self._is_chief:
478 self._is_chief = is_chief486 if self._is_chief:
493 strategy.extended._is_chief = False
51 self._is_chief = is_chief61 elif self._is_chief:
91 self._is_chief = is_chief101 elif self._is_chief:
346 self._is_chief = is_chief947 if self._is_chief:1004 if self._is_chief:
995 is_chief_list = [hook._is_chief for hook in train_ops.train_hooks]