| /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)
|
| 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 | multi_worker_util.py | 75 def is_chief(cluster_spec, task_type, task_id): function
|
| /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/ |
| D | gan_estimator_impl.py | 117 is_chief=True): argument
193 get_hooks_fn, use_loss_summaries, is_chief)
221 get_hooks_fn=None, use_loss_summaries=True, is_chief=True): argument
243 get_hooks_fn, is_chief=is_chief)
328 get_hooks_fn, train_op_fn=tfgan_train.gan_train_ops, is_chief=True): argument
332 discriminator_optimizer, is_chief=is_chief)
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/ |
| D | learning.py | 538 is_chief=True, argument
682 if is_chief:
699 if is_chief and sync_optimizer is not None:
717 if is_chief and trace_every_n_steps is not None:
724 is_chief=is_chief,
754 if is_chief:
764 if is_chief and sync_optimizer is not None:
779 if logdir and sv.is_chief:
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| D | learning.py | 69 is_chief=True, argument
173 is_chief=is_chief,
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | run_config_test.py | 57 self.assertTrue(config.is_chief)
80 self.assertFalse(config.is_chief)
105 self.assertFalse(config.is_chief)
180 self.assertTrue(config.is_chief)
200 self.assertTrue(config.is_chief)
218 self.assertFalse(config.is_chief)
225 self.assertTrue(config.is_chief)
|
| D | kmeans.py | 87 def __init__(self, init_op, is_initialized_op, is_chief): argument
89 self._is_chief = is_chief
141 init_op, is_initialized, config.is_chief)]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| D | model.py | 117 is_chief=config.is_chief,
302 is_chief=config.is_chief,
324 is_chief=config.is_chief,
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | training.py | 479 is_chief=True, argument
523 if logdir is None and is_chief:
534 is_chief=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)
729 supervisor.Supervisor(logdir=logdir, is_chief=False)
732 supervisor.Supervisor(logdir=logdir, is_chief=False)
[all …]
|
| D | sync_replicas_optimizer.py | 461 def make_session_run_hook(self, is_chief, num_tokens=-1): argument
463 return _SyncReplicasOptimizerHook(self, is_chief, num_tokens)
469 def __init__(self, sync_optimizer, is_chief, num_tokens): argument
478 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,
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | model_average_optimizer.py | 115 is_chief, argument
135 self._is_chief = is_chief
298 def __init__(self, ma_optimizer, is_chief): argument
306 self._is_chief = is_chief
|
| D | agn_optimizer.py | 234 def make_session_run_hook(self, is_chief, task_index): argument
236 return _AGNOptimizerHook(self, is_chief, task_index)
241 def __init__(self, agn_optimizer, is_chief, task_index): argument
250 self._is_chief = is_chief
|
| D | model_average_optimizer_test.py | 67 is_chief = (worker_id == 0)
95 is_chief=is_chief,
|
| D | elastic_average_optimizer.py | 396 def make_session_run_hook(self, is_chief, task_index): argument
398 return _ElasticAverageOptimizerHook(self, is_chief, task_index)
448 def __init__(self, ea_optimizer, is_chief, task_index): argument
456 self._is_chief = is_chief
|
| D | agn_optimizer_test.py | 70 is_chief = (worker_id == 0)
121 hook = opt.make_session_run_hook(is_chief, worker_id)
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | gmm.py | 49 def __init__(self, init_op, is_initialized_op, is_chief): argument
51 self._is_chief = is_chief
170 init_op, is_initialized, config.is_chief)]
|
| D | kmeans.py | 79 def __init__(self, init_op, is_initialized_var, is_chief): argument
91 self._is_chief = is_chief
216 _InitializeClustersHook(init_op, is_initialized, config.is_chief)
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch_test.py | 230 is_chief=True,
338 is_chief=True,
494 is_chief=True,
598 is_chief=True,
702 is_chief=True,
774 is_chief=False,
838 is_chief=False,
923 is_chief=False,
1027 is_chief=False,
1075 is_chief=True,
[all …]
|
| /external/tensorflow/tensorflow/contrib/eager/python/ |
| D | parameter_server.py | 239 def parameter_server_scope(is_chief, ps_job_name, num_ps_tasks): argument
273 kwargs["initialize"] = is_chief
278 if not is_chief:
|
| /external/tensorflow/tensorflow/contrib/gan/python/ |
| D | train.py | 929 is_chief=True, argument
1006 sync_hooks.append(generator_optimizer.make_session_run_hook(is_chief))
1028 sync_hooks.append(discriminator_optimizer.make_session_run_hook(is_chief))
1151 is_chief=True, argument
1194 is_chief=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=[\'…
|