aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/training/sync_replicas_optimizer.py
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2016-12-16 10:42:00 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-12-16 11:03:19 -0800
commite6a9b7d61745127ddabccae5fb589f99144a1937 (patch)
tree73f2c54083c66b53b01ab78608b9c2ce90dafa9f /tensorflow/python/training/sync_replicas_optimizer.py
parent06df2b5828f788879988782ad4b23330fae562d5 (diff)
Add names the queues in SyncReplicasOptimizerV2 for easier debugging.
Change: 142274509
Diffstat (limited to 'tensorflow/python/training/sync_replicas_optimizer.py')
-rw-r--r--tensorflow/python/training/sync_replicas_optimizer.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/python/training/sync_replicas_optimizer.py b/tensorflow/python/training/sync_replicas_optimizer.py
index 14b0d10685..1ded30f592 100644
--- a/tensorflow/python/training/sync_replicas_optimizer.py
+++ b/tensorflow/python/training/sync_replicas_optimizer.py
@@ -310,6 +310,7 @@ class SyncReplicasOptimizerV2(optimizer.Optimizer):
data_flow_ops.FIFOQueue(-1,
global_step.dtype.base_dtype,
shapes=(),
+ name="sync_token_q",
shared_name="sync_token_q"))
self._sync_token_queue = sync_token_queue
@@ -320,6 +321,7 @@ class SyncReplicasOptimizerV2(optimizer.Optimizer):
data_flow_ops.FIFOQueue(1,
types_pb2.DT_INT32,
shapes=(),
+ name="dummy_queue",
shared_name="dummy_queue"))
with ops.device(global_step.device), ops.name_scope(""):