diff options
author | 2016-12-16 10:42:00 -0800 | |
---|---|---|
committer | 2016-12-16 11:03:19 -0800 | |
commit | e6a9b7d61745127ddabccae5fb589f99144a1937 (patch) | |
tree | 73f2c54083c66b53b01ab78608b9c2ce90dafa9f /tensorflow/python/training/sync_replicas_optimizer.py | |
parent | 06df2b5828f788879988782ad4b23330fae562d5 (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.py | 2 |
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(""): |