tf.train.add_queue_runner(qr, collection=tf.GraphKeys.QUEUE_RUNNERS)
tf.train.queue_runner.add_queue_runner(qr, collection=tf.GraphKeys.QUEUE_RUNNERS)
See the guide: Training > Coordinator and QueueRunner
Adds a QueueRunner
to a collection in the graph.
When building a complex model that uses many queues it is often difficult to gather all the queue runners that need to be run. This convenience function allows you to add a queue runner to a well known collection in the graph.
The companion method start_queue_runners()
can be used to start threads for all the collected queue runners.
qr
: A QueueRunner
.collection
: A GraphKey
specifying the graph collection to add the queue runner to. Defaults to GraphKeys.QUEUE_RUNNERS
.Defined in tensorflow/python/training/queue_runner_impl.py
.
© 2017 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/train/add_queue_runner