Home
last modified time | relevance | path

Searched refs:QueueRunner (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/cc/training/
Dqueue_runner.cc22 Status QueueRunner::New(const QueueRunnerDef& queue_runner_def, in New()
23 std::unique_ptr<QueueRunner>* result) { in New()
24 result->reset(new QueueRunner()); in New()
28 Status QueueRunner::New(const QueueRunnerDef& queue_runner_def, in New()
30 std::unique_ptr<QueueRunner>* result) { in New()
31 result->reset(new QueueRunner()); in New()
36 void QueueRunner::AddErrorCallback(const std::function<void(Status)>& cb) { in AddErrorCallback()
41 void QueueRunner::ClearErrorCallbacks() { in ClearErrorCallbacks()
46 Status QueueRunner::Init(const QueueRunnerDef& queue_runner_def) { in Init()
82 QueueRunner::~QueueRunner() { in ~QueueRunner()
[all …]
Dqueue_runner_test.cc118 std::unique_ptr<QueueRunner> qr; in TEST()
119 TF_EXPECT_OK(QueueRunner::New(queue_runner_def, &qr)); in TEST()
138 std::unique_ptr<QueueRunner> qr; in TEST()
139 TF_EXPECT_OK(QueueRunner::New(queue_runner_def, &qr)); in TEST()
157 std::unique_ptr<QueueRunner> qr; in TEST()
158 TF_EXPECT_OK(QueueRunner::New(queue_runner_def, &qr)); in TEST()
171 std::unique_ptr<QueueRunner> qr; in TEST()
172 TF_EXPECT_OK(QueueRunner::New(queue_runner_def, &qr)); in TEST()
210 std::unique_ptr<QueueRunner> qr; in TEST()
211 TF_EXPECT_OK(QueueRunner::New(queue_runner_def, &qr)); in TEST()
[all …]
Dqueue_runner.h38 class QueueRunner : public RunnerInterface {
44 std::unique_ptr<QueueRunner>* result);
48 std::unique_ptr<QueueRunner>* result);
57 ~QueueRunner();
89 QueueRunner() : coord_(nullptr), stopped_(false), cg_mu_(nullptr) {} in QueueRunner() function
/external/tensorflow/tensorflow/python/training/
Dqueue_runner_test.py53 qr = queue_runner_impl.QueueRunner(queue, [count_up_to])
74 qr = queue_runner_impl.QueueRunner(queue, [count_up_to_3, count_up_to_30])
91 qr = queue_runner_impl.QueueRunner(queue, [_MockOp("i fail"),
112 qr = queue_runner_impl.QueueRunner(q1, [enqueue1])
140 qr = queue_runner_impl.QueueRunner(queue, [count_up_to])
159 qr = queue_runner_impl.QueueRunner(queue, [_MockOp("not an op")])
174 qr = queue_runner_impl.QueueRunner(queue, [enqueue])
194 qr = queue_runner_impl.QueueRunner(queue, [count_up_to])
209 qr = queue_runner_impl.QueueRunner(queue, [count_up_to])
224 qr = queue_runner_impl.QueueRunner(queue, [count_up_to,
[all …]
Dqueue_runner_impl.py38 class QueueRunner(object): class
390 return QueueRunner(queue_runner_def=queue_runner_def,
485 to_proto=QueueRunner.to_proto,
486 from_proto=QueueRunner.from_proto)
Dinput.py198 queue_runner.QueueRunner(
736 queue_runner.add_queue_runner(queue_runner.QueueRunner(queue, enqueue_ops))
753 queue_runner.add_queue_runner(queue_runner.QueueRunner(queue, enqueue_ops))
Dsync_replicas_optimizer.py350 self._chief_queue_runner = queue_runner.QueueRunner(dummy_queue,
Dsaver_test.py1636 qr = queue_runner_impl.QueueRunner(input_queue, [count_up_to])
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.-queue-runner.pbtxt1 path: "tensorflow.train.QueueRunner"
3 is_instance: "<class \'tensorflow.python.training.queue_runner_impl.QueueRunner\'>"
Dtensorflow.train.queue_runner.-queue-runner.pbtxt1 path: "tensorflow.train.queue_runner.QueueRunner"
3 is_instance: "<class \'tensorflow.python.training.queue_runner_impl.QueueRunner\'>"
Dtensorflow.train.queue_runner.pbtxt4 name: "QueueRunner"
Dtensorflow.train.pbtxt152 name: "QueueRunner"
/external/tensorflow/tensorflow/core/protobuf/
Dqueue_runner.proto11 // Protocol buffer representing a QueueRunner.
/external/tensorflow/tensorflow/contrib/slim/python/slim/data/
Dprefetch_queue.py89 queue_runner.QueueRunner(queue, [enqueue_op] * num_threads))
Dparallel_reader.py165 queue_runner.QueueRunner(self._common_queue, enqueue_ops))
/external/tensorflow/tensorflow/contrib/training/python/training/
Dbucket_ops.py269 queue_runner.QueueRunner(
276 queue_runner.QueueRunner(
Dsampling_ops_threading_test.py58 queue_runner_impl.QueueRunner(queue, [enqueue_op] * 25))
Dsequence_queueing_state_saver.py1531 q_runner = queue_runner.QueueRunner(
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine.cc174 std::unique_ptr<QueueRunner> queue_runner; in Run()
175 TF_RETURN_IF_ERROR(QueueRunner::New(queue_runner_defs_[i], in Run()
/external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/
Dgraph_io.py416 queue_runner.QueueRunner(file_name_queue, [enqueue_op]))
719 queue_runner.QueueRunner(
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dgmm_test.py207 queue_runner.add_queue_runner(queue_runner.QueueRunner(queue,
Dkmeans_test.py601 queue_runner.QueueRunner(queue, [enqueue_op]))
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Destimator_input_test.py63 queue_runner = queue_runner_impl.QueueRunner(fake_queue,
Dkmeans_test.py570 queue_runner.QueueRunner(queue, [enqueue_op]))
/external/tensorflow/tensorflow/python/framework/
Dmeta_graph_test.py670 qr = queue_runner_impl.QueueRunner(input_queue, [enqueue], close)