Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item.cc57 for (const auto& queue_runner : queue_runners) { in EnqueueOpsFanin() local
58 for (const string& enqueue_op : queue_runner.enqueue_op_name()) { in EnqueueOpsFanin()
106 for (const auto& queue_runner : queue_runners) { in NodesToPreserve() local
107 for (const string& enqueue_op : queue_runner.enqueue_op_name()) { in NodesToPreserve()
110 if (!queue_runner.close_op_name().empty()) { in NodesToPreserve()
111 result.insert(NodeName(queue_runner.close_op_name())); in NodesToPreserve()
113 if (!queue_runner.cancel_op_name().empty()) { in NodesToPreserve()
114 result.insert(NodeName(queue_runner.cancel_op_name())); in NodesToPreserve()
Dgrappler_item_builder.cc418 QueueRunnerDef queue_runner; in GrapplerItemFromMetaGraphDef() local
419 if (!queue_runner.ParseFromString(raw)) { in GrapplerItemFromMetaGraphDef()
423 if (queue_runner.cancel_op_name().empty()) { in GrapplerItemFromMetaGraphDef()
427 new_item->queue_runners.push_back(queue_runner); in GrapplerItemFromMetaGraphDef()
/external/tensorflow/tensorflow/core/grappler/inputs/
Dtrivial_test_graph_input_yielder.cc113 QueueRunnerDef queue_runner; in NextItem() local
114 queue_runner.set_queue_name("queue"); in NextItem()
115 queue_runner.set_cancel_op_name("cancel"); in NextItem()
116 *queue_runner.add_enqueue_op_name() = "enqueue"; in NextItem()
117 r.queue_runners.push_back(queue_runner); in NextItem()
/external/tensorflow/tensorflow/contrib/slim/python/slim/data/
Dprefetch_queue.py26 from tensorflow.python.training import queue_runner
88 queue_runner.add_queue_runner(
89 queue_runner.QueueRunner(queue, [enqueue_op] * num_threads))
Dparallel_reader.py29 from tensorflow.python.training import queue_runner
164 queue_runner.add_queue_runner(
165 queue_runner.QueueRunner(self._common_queue, enqueue_ops))
/external/tensorflow/tensorflow/contrib/training/python/training/
Dbucket_ops.py41 from tensorflow.python.training import queue_runner
268 queue_runner.add_queue_runner(
269 queue_runner.QueueRunner(
275 queue_runner.add_queue_runner(
276 queue_runner.QueueRunner(
/external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/
Dgraph_io.py41 from tensorflow.python.training import queue_runner
415 queue_runner.add_queue_runner(
416 queue_runner.QueueRunner(file_name_queue, [enqueue_op]))
718 queue_runner.add_queue_runner(
719 queue_runner.QueueRunner(
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine.cc174 std::unique_ptr<QueueRunner> queue_runner; in Run() local
176 coordinator_.get(), &queue_runner)); in Run()
178 TF_RETURN_IF_ERROR(queue_runner->StartAndCollectCostGraph( in Run()
181 coordinator_->RegisterRunner(std::move(queue_runner))); in Run()
Dsingle_machine_test.cc527 QueueRunnerDef queue_runner; in CreateGrapplerItemWithResourceMemory() local
528 queue_runner.set_queue_name("queue"); in CreateGrapplerItemWithResourceMemory()
529 *queue_runner.add_enqueue_op_name() = "enqueue"; in CreateGrapplerItemWithResourceMemory()
530 item.queue_runners.push_back(queue_runner); in CreateGrapplerItemWithResourceMemory()
DBUILD116 "//tensorflow/cc:queue_runner",
/external/tensorflow/tensorflow/cc/
DBUILD724 name = "queue_runner",
725 srcs = ["training/queue_runner.cc"],
726 hdrs = ["training/queue_runner.h"],
743 ":queue_runner",
774 ":queue_runner",
798 ":queue_runner",
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dgmm_test.py33 from tensorflow.python.training import queue_runner
207 queue_runner.add_queue_runner(queue_runner.QueueRunner(queue,
Dkmeans_test.py43 from tensorflow.python.training import queue_runner
600 queue_runner.add_queue_runner(
601 queue_runner.QueueRunner(queue, [enqueue_op]))
/external/tensorflow/tensorflow/python/training/
Dinput.py47 from tensorflow.python.training import queue_runner
197 queue_runner.add_queue_runner(
198 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.py32 from tensorflow.python.training import queue_runner
350 self._chief_queue_runner = queue_runner.QueueRunner(dummy_queue,
Dtraining.py49 from tensorflow.python.training.queue_runner import *
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.queue_runner.pbtxt1 path: "tensorflow.train.queue_runner"
Dtensorflow.train.queue_runner.-queue-runner.pbtxt1 path: "tensorflow.train.queue_runner.QueueRunner"
/external/tensorflow/tensorflow/contrib/learn/python/learn/
Dgraph_actions.py51 from tensorflow.python.training import queue_runner
557 threads = queue_runner.start_queue_runners(session, coord)
696 threads = queue_runner.start_queue_runners(session, coord=coord)
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dkmeans_test.py43 from tensorflow.python.training import queue_runner
569 queue_runner.add_queue_runner(
570 queue_runner.QueueRunner(queue, [enqueue_op]))
Destimator_input_test.py63 queue_runner = queue_runner_impl.QueueRunner(fake_queue,
65 queue_runner_impl.add_queue_runner(queue_runner)
/external/tensorflow/tensorflow/contrib/makefile/
Dtf_proto_files.txt42 tensorflow/core/protobuf/queue_runner.proto
Dproto_text_pb_cc_files.txt36 tensorflow/core/protobuf/queue_runner.pb.cc
Dproto_text_pb_h_files.txt36 tensorflow/core/protobuf/queue_runner.pb.h
/external/tensorflow/tensorflow/cc/training/
Dqueue_runner_test.cc231 void JoinThread(QueueRunner* queue_runner, bool* join_succeeded, in JoinThread() argument
233 EXPECT_EQ(queue_runner->Join().code(), Code::CANCELLED); in JoinThread()

12