| /external/tensorflow/tensorflow/cc/training/ |
| queue_runner.cc | 22 Status QueueRunner::New(const QueueRunnerDef& queue_runner_def,
23 std::unique_ptr<QueueRunner>* result) {
24 result->reset(new QueueRunner());
28 Status QueueRunner::New(const QueueRunnerDef& queue_runner_def,
30 std::unique_ptr<QueueRunner>* result) {
31 result->reset(new QueueRunner());
36 void QueueRunner::AddErrorCallback(const std::function<void(Status)>& cb) {
41 void QueueRunner::ClearErrorCallbacks() {
46 Status QueueRunner::Init(const QueueRunnerDef& queue_runner_def) {
82 QueueRunner::~QueueRunner()
[all...] |
| queue_runner.h | 36 /// QueueRunner class imitates the behavior of the python version of QueueRunner
38 class QueueRunner : public RunnerInterface {
40 /// Creates a new QueueRunner from proto.
44 std::unique_ptr<QueueRunner>* result);
46 /// Creates a new QueueRunner with a coordinator, see coordinator.h for usage.
48 std::unique_ptr<QueueRunner>* result);
57 ~QueueRunner();
89 QueueRunner() : coord_(nullptr), stopped_(false), cg_mu_(nullptr) {}
|
| queue_runner_test.cc | 118 std::unique_ptr<QueueRunner> qr;
119 TF_EXPECT_OK(QueueRunner::New(queue_runner_def, &qr));
138 std::unique_ptr<QueueRunner> qr;
139 TF_EXPECT_OK(QueueRunner::New(queue_runner_def, &qr));
157 std::unique_ptr<QueueRunner> qr;
158 TF_EXPECT_OK(QueueRunner::New(queue_runner_def, &qr));
171 std::unique_ptr<QueueRunner> qr;
172 TF_EXPECT_OK(QueueRunner::New(queue_runner_def, &qr));
210 std::unique_ptr<QueueRunner> qr;
211 TF_EXPECT_OK(QueueRunner::New(queue_runner_def, &qr))
[all...] |
| /external/tensorflow/tensorflow/python/training/ |
| queue_runner_test.py | 15 """Tests for QueueRunner."""
51 qr = queue_runner_impl.QueueRunner(queue, [count_up_to])
72 qr = queue_runner_impl.QueueRunner(queue, [count_up_to_3, count_up_to_30])
89 qr = queue_runner_impl.QueueRunner(queue, [_MockOp("i fail"),
110 qr = queue_runner_impl.QueueRunner(q1, [enqueue1])
138 qr = queue_runner_impl.QueueRunner(queue, [count_up_to])
157 qr = queue_runner_impl.QueueRunner(queue, [_MockOp("not an op")])
172 qr = queue_runner_impl.QueueRunner(queue, [enqueue])
192 qr = queue_runner_impl.QueueRunner(queue, [count_up_to])
207 qr = queue_runner_impl.QueueRunner(queue, [count_up_to]
[all...] |
| queue_runner_impl.py | 33 @tf_export("train.queue_runner.QueueRunner", "train.QueueRunner")
34 class QueueRunner(object):
48 The `QueueRunner`, combined with the `Coordinator`, helps handle these issues.
59 """Create a QueueRunner.
61 On construction the `QueueRunner` adds an op to close the queue. That op
64 When you later call the `create_threads()` method, the `QueueRunner` will
81 recreates the QueueRunner from its contents. `queue_runner_def` and the
117 """Create a QueueRunner from arguments.
164 """Create a QueueRunner from `QueueRunnerDef`
[all...] |
| input.py | 149 A queue with the output rows. A `QueueRunner` for the queue is
185 queue_runner.QueueRunner(
227 A queue with the output strings. A `QueueRunner` for the Queue
286 A Queue with the output integers. A `QueueRunner` for the Queue
306 Implemented using a Queue -- a `QueueRunner` for the Queue
710 queue_runner.add_queue_runner(queue_runner.QueueRunner(queue, enqueue_ops))
727 queue_runner.add_queue_runner(queue_runner.QueueRunner(queue, enqueue_ops))
[all...] |
| sync_replicas_optimizer.py | 334 self._chief_queue_runner = queue_runner.QueueRunner(dummy_queue,
346 """Returns the QueueRunner for the chief to execute.
352 actually generates this queuerunner.
355 A `QueueRunner` for chief to execute.
|
| saver_test.py | [all...] |
| /external/tensorflow/tensorflow/contrib/slim/python/slim/data/ |
| prefetch_queue.py | 88 queue_runner.QueueRunner(queue, [enqueue_op] * num_threads))
|
| parallel_reader.py | 165 queue_runner.QueueRunner(self._common_queue, enqueue_ops))
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| sampling_ops_threading_test.py | 58 queue_runner_impl.QueueRunner(queue, [enqueue_op] * 25))
|
| bucket_ops.py | 88 This function is implemented using several queues. A `QueueRunner` for the
269 queue_runner.QueueRunner(
276 queue_runner.QueueRunner(
|
| sequence_queueing_state_saver.py | 624 `QueueRunner` (see the example below).
707 queue_runner = tf.train.QueueRunner(
717 **Note**: Usually the barrier is given to a QueueRunner as in the
718 examples above. The QueueRunner will close the barrier if the prefetch_op
[all...] |
| /external/tensorflow/tensorflow/python/estimator/inputs/queues/ |
| feeding_queue_runner.py | 16 """A `QueueRunner` that takes a feed function as an argument."""
29 class _FeedingQueueRunner(qr.QueueRunner):
84 # TODO(jamieas): Reduce code duplication with `QueueRunner`.
112 logging.error("Exception in QueueRunner: %s", str(e))
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| single_machine.cc | 161 std::unique_ptr<QueueRunner> queue_runner;
162 TF_RETURN_IF_ERROR(QueueRunner::New(queue_runner_defs_[i],
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/ |
| graph_io.py | 407 queue_runner.QueueRunner(file_name_queue, [enqueue_op]))
699 # Use a single QueueRunner with multiple threads to enqueue so the queue is
707 queue_runner.QueueRunner(
[all...] |
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| gmm_test.py | 207 queue_runner.add_queue_runner(queue_runner.QueueRunner(queue,
212 # This test makes sure that there are no deadlocks when using a QueueRunner.
214 # is generated using a QueueRunner, one has to make sure that these runners
|
| kmeans_test.py | 563 queue_runner.QueueRunner(queue, [enqueue_op]))
568 # This test makes sure that there are no deadlocks when using a QueueRunner.
570 # is generated using a QueueRunner, one has to make sure that these runners
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| estimator_input_test.py | 63 queue_runner = queue_runner_impl.QueueRunner(fake_queue,
|
| kmeans_test.py | 571 queue_runner.QueueRunner(queue, [enqueue_op]))
576 # This test makes sure that there are no deadlocks when using a QueueRunner.
578 # is generated using a QueueRunner, one has to make sure that these runners
|
| /external/tensorflow/tensorflow/python/estimator/canned/ |
| dnn_test.py | 152 queue_runner.QueueRunner(
|
| baseline_test.py | 106 queue_runner.QueueRunner(input_queue,
[all...] |
| linear_testing_utils.py | 117 queue_runner.QueueRunner(input_queue,
[all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| meta_graph_test.py | 620 qr = queue_runner_impl.QueueRunner(input_queue, [enqueue], close)
[all...] |
