| /external/tensorflow/tensorflow/python/distribute/ |
| D | multi_worker_util_test.py | 85 self.assertTrue(multi_worker_util.is_chief(cluster_spec, "chief", 0))
86 self.assertFalse(multi_worker_util.is_chief(cluster_spec, "worker", 0))
90 self.assertTrue(multi_worker_util.is_chief(cluster_spec, "worker", 0))
91 self.assertFalse(multi_worker_util.is_chief(cluster_spec, "worker", 1))
95 multi_worker_util.is_chief(cluster_spec, "chief", 0)
99 multi_worker_util.is_chief(cluster_spec, "worker", 2)
107 self.assertTrue(multi_worker_util.is_chief(cluster_spec, "evaluator", 0))
|
| D | estimator_training.py | 96 config._is_chief = worker_context.is_chief
236 dc.CoordinatorMode.INDEPENDENT_WORKER or context.is_chief):
325 if context.is_chief:
374 if context.is_chief:
|
| D | multi_worker_util.py | 108 def is_chief(cluster_spec=None, task_type=None, task_id=None): function
133 return dc_context.get_current_worker_context().is_chief
|
| D | distribute_coordinator.py | 297 def is_chief(self): member in _WorkerContext
861 if context.is_chief:
|
| D | distribute_coordinator_test.py | 285 if context.is_chief:
369 context.is_chief,
|
| D | parameter_server_strategy.py | 265 self._is_chief = multi_worker_util.is_chief(cluster_spec, task_type,
|
| D | multi_worker_test_base.py | 837 def is_chief(): function
|
| D | collective_all_reduce_strategy.py | 471 self._is_chief = multi_worker_util.is_chief(cluster_spec, task_type,
|
| /external/tensorflow/tensorflow/python/training/ |
| D | supervisor_test.py | 345 sv = supervisor.Supervisor(is_chief=True, logdir=logdir, summary_op=None)
392 sv = supervisor.Supervisor(is_chief=False)
401 sv = supervisor.Supervisor(is_chief=False)
550 def get_session(is_chief): argument
566 is_chief=is_chief,
608 def get_session(is_chief): argument
624 is_chief=is_chief,
683 sv = supervisor.Supervisor(logdir=logdir, init_op=None, is_chief=False)
722 supervisor.Supervisor(logdir=logdir, is_chief=False)
725 supervisor.Supervisor(logdir=logdir, is_chief=False)
[all …]
|
| D | sync_replicas_optimizer.py | 446 def make_session_run_hook(self, is_chief, num_tokens=-1): argument
448 return _SyncReplicasOptimizerHook(self, is_chief, num_tokens)
454 def __init__(self, sync_optimizer, is_chief, num_tokens): argument
463 self._is_chief = is_chief
|
| D | sync_replicas_optimizer_test.py | 38 is_chief = (worker_id == 0)
69 is_chief, num_tokens=num_workers)
74 is_chief=is_chief,
|
| D | monitored_session_test.py | 294 is_chief=True, checkpoint_dir=logdir) as session:
300 is_chief=True, checkpoint_dir=logdir) as session:
309 is_chief=True,
317 is_chief=True, checkpoint_dir=logdir) as session:
326 is_chief=True,
336 is_chief=True, checkpoint_dir=logdir) as session:
385 is_chief=True,
403 is_chief=True,
424 is_chief=True,
436 is_chief=True, checkpoint_dir=logdir) as session:
[all …]
|
| D | supervisor.py | 209 is_chief=True, argument
321 self._is_chief = is_chief
499 def is_chief(self): member in Supervisor
|
| D | monitored_session.py | 334 if chief_only_hooks and worker_context.is_chief:
430 is_chief=True, argument
555 if not is_chief:
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.-supervisor.pbtxt | 26 name: "is_chief"
107 …argspec: "args=[\'self\', \'graph\', \'ready_op\', \'ready_for_local_init_op\', \'is_chief\', \'in…
|
| D | tensorflow.train.-sync-replicas-optimizer.pbtxt | 53 …argspec: "args=[\'self\', \'is_chief\', \'num_tokens\'], varargs=None, keywords=None, defaults=[\'…
|
| D | tensorflow.estimator.tpu.-run-config.pbtxt | 39 name: "is_chief"
|
| D | tensorflow.estimator.-run-config.pbtxt | 34 name: "is_chief"
|
| D | tensorflow.train.pbtxt | 257 …argspec: "args=[\'master\', \'is_chief\', \'checkpoint_dir\', \'scaffold\', \'hooks\', \'chief_onl…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.estimator.-run-config.pbtxt | 34 name: "is_chief"
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | distribute_coordinator_utils.py | 262 def is_chief(self): member in _WorkerContext
641 if context.is_chief:
|
| D | distributed_training_utils_v1.py | 1084 return dc.get_current_worker_context().is_chief
|
| /external/tensorflow/tensorflow/python/distribute/failure_handling/ |
| D | failure_handler_test.py | 65 multi_worker_util.is_chief(
|
| D | failure_handling.py | 437 self._is_chief = multi_worker_util.is_chief(
|
| D | gce_failure_handler_test.py | 66 multi_worker_util.is_chief(
|