Home
last modified time | relevance | path

Searched refs:is_chief (Results 1 – 25 of 48) sorted by relevance

12

/external/tensorflow/tensorflow/python/distribute/
Dmulti_worker_util_test.py89 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)
Destimator_training.py100 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:
Dmulti_worker_util.py75 def is_chief(cluster_spec, task_type, task_id): function
/external/tensorflow/tensorflow/contrib/gan/python/estimator/python/
Dgan_estimator_impl.py117 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/
Dlearning.py538 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/
Dlearning.py69 is_chief=True, argument
173 is_chief=is_chief,
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Drun_config_test.py57 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)
Dkmeans.py87 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/
Dmodel.py117 is_chief=config.is_chief,
302 is_chief=config.is_chief,
324 is_chief=config.is_chief,
/external/tensorflow/tensorflow/contrib/training/python/training/
Dtraining.py479 is_chief=True, argument
523 if logdir is None and is_chief:
534 is_chief=is_chief,
/external/tensorflow/tensorflow/python/training/
Dsupervisor_test.py350 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 …]
Dsync_replicas_optimizer_test.py41 is_chief = (worker_id == 0)
72 is_chief, num_tokens=num_workers)
77 is_chief=is_chief,
Dsync_replicas_optimizer.py461 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
/external/tensorflow/tensorflow/contrib/opt/python/training/
Dmodel_average_optimizer.py115 is_chief, argument
135 self._is_chief = is_chief
298 def __init__(self, ma_optimizer, is_chief): argument
306 self._is_chief = is_chief
Dagn_optimizer.py234 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
Dmodel_average_optimizer_test.py67 is_chief = (worker_id == 0)
95 is_chief=is_chief,
Delastic_average_optimizer.py396 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
Dagn_optimizer_test.py70 is_chief = (worker_id == 0)
121 hook = opt.make_session_run_hook(is_chief, worker_id)
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dgmm.py49 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)]
Dkmeans.py79 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/
Dgbdt_batch_test.py230 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/
Dparameter_server.py239 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/
Dtrain.py929 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/
Dtensorflow.train.-supervisor.pbtxt26 name: "is_chief"
107 …argspec: "args=[\'self\', \'graph\', \'ready_op\', \'ready_for_local_init_op\', \'is_chief\', \'in…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.estimator.-run-config.pbtxt26 name: "is_chief"

12