| /external/tensorflow/tensorflow/python/distribute/ |
| D | multi_worker_util_test.py | 89 self.assertTrue(multi_worker_util.is_chief(cluster_spec, "chief", 0))
90 self.assertFalse(multi_worker_util.is_chief(cluster_spec, "worker", 0))
94 self.assertTrue(multi_worker_util.is_chief(cluster_spec, "worker", 0))
95 self.assertFalse(multi_worker_util.is_chief(cluster_spec, "worker", 1))
99 multi_worker_util.is_chief(cluster_spec, "chief", 0)
103 multi_worker_util.is_chief(cluster_spec, "worker", 2)
111 self.assertTrue(multi_worker_util.is_chief(cluster_spec, "evaluator", 0))
|
| D | multi_worker_util.py | 112 def is_chief(cluster_spec=None, task_type=None, task_id=None): function
137 return dc_context.get_current_worker_context().is_chief
|
| D | estimator_training.py | 100 config._is_chief = worker_context.is_chief
240 dc.CoordinatorMode.INDEPENDENT_WORKER or context.is_chief):
329 if context.is_chief:
378 if context.is_chief:
|
| D | distribute_coordinator.py | 301 def is_chief(self): member in _WorkerContext
865 if context.is_chief:
|
| D | distribute_coordinator_test.py | 299 if context.is_chief:
383 context.is_chief,
|
| D | parameter_server_strategy.py | 267 self._is_chief = multi_worker_util.is_chief(cluster_spec, task_type,
|
| D | collective_all_reduce_strategy.py | 427 self._is_chief = multi_worker_util.is_chief(cluster_spec, task_type,
|
| D | multi_worker_test_base.py | 859 def is_chief(): function
|
| D | parameter_server_strategy_test.py | 499 multi_worker_util.is_chief(
|
| /external/tensorflow/tensorflow/python/training/ |
| D | supervisor_test.py | 350 sv = supervisor.Supervisor(is_chief=True, logdir=logdir, summary_op=None)
397 sv = supervisor.Supervisor(is_chief=False)
406 sv = supervisor.Supervisor(is_chief=False)
555 def get_session(is_chief): argument
571 is_chief=is_chief,
613 def get_session(is_chief): argument
629 is_chief=is_chief,
688 sv = supervisor.Supervisor(logdir=logdir, init_op=None, is_chief=False)
727 supervisor.Supervisor(logdir=logdir, is_chief=False)
730 supervisor.Supervisor(logdir=logdir, is_chief=False)
[all …]
|
| D | sync_replicas_optimizer.py | 449 def make_session_run_hook(self, is_chief, num_tokens=-1): argument
451 return _SyncReplicasOptimizerHook(self, is_chief, num_tokens)
457 def __init__(self, sync_optimizer, is_chief, num_tokens): argument
466 self._is_chief = is_chief
|
| D | sync_replicas_optimizer_test.py | 41 is_chief = (worker_id == 0)
72 is_chief, num_tokens=num_workers)
77 is_chief=is_chief,
|
| D | monitored_session_test.py | 296 is_chief=True, checkpoint_dir=logdir) as session:
302 is_chief=True, checkpoint_dir=logdir) as session:
311 is_chief=True,
319 is_chief=True, checkpoint_dir=logdir) as session:
328 is_chief=True,
338 is_chief=True, checkpoint_dir=logdir) as session:
348 is_chief=True,
366 is_chief=True,
387 is_chief=True,
399 is_chief=True, checkpoint_dir=logdir) as session:
[all …]
|
| D | supervisor.py | 213 is_chief=True, argument
325 self._is_chief = is_chief
503 def is_chief(self): member in Supervisor
|
| D | monitored_session.py | 340 if chief_only_hooks and worker_context.is_chief:
436 is_chief=True, argument
550 if not is_chief:
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | multi_worker_callback_tf2_test.py | 130 checkpoint_exists(saving_filepath), test_base.is_chief())
140 test_base.is_chief())
271 test_base.is_chief())
|
| D | distributed_training_utils_v1.py | 1094 return dc_context.get_current_worker_context().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.-run-config.pbtxt | 34 name: "is_chief"
|
| D | tensorflow.estimator.tpu.-run-config.pbtxt | 39 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"
|