HomeSort by relevance Sort by last modified time
    Searched refs:sync_optimizer (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/contrib/model_pruning/python/
learning.py 80 sync_optimizer=None,
86 When the sync_optimizer is supplied, gradient updates are applied
132 that this must be 0 if a sync_optimizer is supplied.
136 sync_optimizer: an instance of tf.train.SyncReplicasOptimizer, or a list of
150 non-zero when `sync_optimizer` is supplied, if `number_of_steps` is
184 sync_optimizer=sync_optimizer,
  /external/tensorflow/tensorflow/contrib/slim/python/slim/
learning.py 552 sync_optimizer=None,
559 When the sync_optimizer is supplied, gradient updates are applied
603 that this must be 0 if a sync_optimizer is supplied.
607 sync_optimizer: an instance of tf.train.SyncReplicasOptimizer, or a list of
628 non-zero when `sync_optimizer` is supplied, if `number_of_steps` is
644 if isinstance(sync_optimizer, sync_replicas_optimizer.SyncReplicasOptimizer):
645 sync_optimizer = [sync_optimizer]
646 if sync_optimizer is not None and startup_delay_steps > 0:
648 'startup_delay_steps must be zero when sync_optimizer is supplied.'
    [all...]
  /external/tensorflow/tensorflow/contrib/kfac/examples/
convnet.py 290 sync_optimizer = tf.train.SyncReplicasOptimizer(
293 train_op = sync_optimizer.minimize(loss, global_step=global_step)
297 hooks = [sync_optimizer.make_session_run_hook(is_chief)]
  /external/tensorflow/tensorflow/python/training/
sync_replicas_optimizer.py 453 def __init__(self, sync_optimizer, is_chief, num_tokens):
457 sync_optimizer: `SyncReplicasOptimizer` which this hook will initialize.
461 self._sync_optimizer = sync_optimizer
  /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
dnn_linear_combined_test.py 282 sync_optimizer = sync_replicas_optimizer.SyncReplicasOptimizer(
286 sync_hook = sync_optimizer.make_session_run_hook(is_chief=True)
291 dnn_optimizer=sync_optimizer)
    [all...]

Completed in 341 milliseconds