diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2016-11-17 12:24:47 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-11-17 12:45:03 -0800 |
| commit | 4dcd9c50c7ba359e3d31a7d0242f7b89387fc2aa (patch) | |
| tree | 538cf2e9db932d6d83ecd7cfeaf2dac4bfea1579 /tensorflow/python/training/sync_replicas_optimizer.py | |
| parent | 61229c29924f216c1954242de483cd0a5acc9fdd (diff) | |
Another missed _ref in sync_replicas_optimizer.
Change: 139491940
Diffstat (limited to 'tensorflow/python/training/sync_replicas_optimizer.py')
| -rw-r--r-- | tensorflow/python/training/sync_replicas_optimizer.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/training/sync_replicas_optimizer.py b/tensorflow/python/training/sync_replicas_optimizer.py index 0940560160..627445215d 100644 --- a/tensorflow/python/training/sync_replicas_optimizer.py +++ b/tensorflow/python/training/sync_replicas_optimizer.py @@ -955,7 +955,7 @@ class SyncReplicasOptimizer(optimizer.Optimizer): if num_tokens > 0: with ops.device(self._global_step.device), ops.name_scope(""): tokens = array_ops.fill([num_tokens], - self._global_step.ref()) + self._global_step._ref()) # pylint: disable=protected-access init_tokens = self._sync_token_queue.enqueue_many((tokens,)) else: init_tokens = control_flow_ops.no_op(name="no_init_tokens") |