diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-08-02 23:56:27 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-08-02 23:59:49 -0700 |
commit | 8ad9557629c5570a2faec9f92167ba94ae33f80f (patch) | |
tree | 076ef9eb2414bda9cc55a761438b95f66227360b /tensorflow/python/training/queue_runner_impl.py | |
parent | e7cfa68edd4eab966986a4c14723443f2c4c6b78 (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.py | 12 |
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) |