aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/training/queue_runner_impl.py
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-08-02 23:56:27 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-08-02 23:59:49 -0700
commit8ad9557629c5570a2faec9f92167ba94ae33f80f (patch)
tree076ef9eb2414bda9cc55a761438b95f66227360b /tensorflow/python/training/queue_runner_impl.py
parente7cfa68edd4eab966986a4c14723443f2c4c6b78 (diff)
Give queue runner threads sensible names.
PiperOrigin-RevId: 164087209
Diffstat (limited to 'tensorflow/python/training/queue_runner_impl.py')
-rw-r--r--tensorflow/python/training/queue_runner_impl.py12
1 files changed, 9 insertions, 3 deletions
diff --git a/tensorflow/python/training/queue_runner_impl.py b/tensorflow/python/training/queue_runner_impl.py
index 4e58602a6f..5abc6a2f58 100644
--- a/tensorflow/python/training/queue_runner_impl.py
+++ b/tensorflow/python/training/queue_runner_impl.py
@@ -316,11 +316,17 @@ class QueueRunner(object):
self._runs_per_session[sess] = len(self._enqueue_ops)
self._exceptions_raised = []
- ret_threads = [threading.Thread(target=self._run, args=(sess, op, coord))
- for op in self._enqueue_ops]
+ ret_threads = []
+ for op in self._enqueue_ops:
+ name = "QueueRunnerThread-{}-{}".format(self.name, op.name)
+ ret_threads.append(threading.Thread(target=self._run,
+ args=(sess, op, coord),
+ name=name))
if coord:
+ name = "QueueRunnerThread-{}-close_on_stop".format(self.name)
ret_threads.append(threading.Thread(target=self._close_on_stop,
- args=(sess, self._cancel_op, coord)))
+ args=(sess, self._cancel_op, coord),
+ name=name))
for t in ret_threads:
if coord:
coord.register_thread(t)