Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item.cc78 for (const auto& queue_runner : queue_runners) { in EnqueueOpsFanin() local
79 for (const string& enqueue_op : queue_runner.enqueue_op_name()) { in EnqueueOpsFanin()
132 for (const auto& queue_runner : queue_runners) { in NodesToPreserve() local
133 for (const string& enqueue_op : queue_runner.enqueue_op_name()) { in NodesToPreserve()
136 if (!queue_runner.close_op_name().empty()) { in NodesToPreserve()
137 result.insert(NodeName(queue_runner.close_op_name())); in NodesToPreserve()
139 if (!queue_runner.cancel_op_name().empty()) { in NodesToPreserve()
140 result.insert(NodeName(queue_runner.cancel_op_name())); in NodesToPreserve()
Dgrappler_item_builder.cc526 QueueRunnerDef queue_runner; in GrapplerItemFromMetaGraphDef() local
527 if (!queue_runner.ParseFromString(raw)) { in GrapplerItemFromMetaGraphDef()
531 if (queue_runner.cancel_op_name().empty()) { in GrapplerItemFromMetaGraphDef()
535 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/core/grappler/clusters/
Dsingle_machine.cc175 std::unique_ptr<QueueRunner> queue_runner; in Run() local
177 coordinator_.get(), &queue_runner)); in Run()
179 TF_RETURN_IF_ERROR(queue_runner->StartAndCollectCostGraph(session_.get(), in Run()
181 TF_RETURN_IF_ERROR(coordinator_->RegisterRunner(std::move(queue_runner))); in Run()
Dsingle_machine_test.cc541 QueueRunnerDef queue_runner; in CreateGrapplerItemWithResourceMemory() local
542 queue_runner.set_queue_name("queue"); in CreateGrapplerItemWithResourceMemory()
543 *queue_runner.add_enqueue_op_name() = "enqueue"; in CreateGrapplerItemWithResourceMemory()
544 item.queue_runners.push_back(queue_runner); in CreateGrapplerItemWithResourceMemory()
DBUILD130 "//tensorflow/cc:queue_runner",
/external/tensorflow/tensorflow/python/training/
Dinput.py40 from tensorflow.python.training import queue_runner
191 queue_runner.add_queue_runner(
192 queue_runner.QueueRunner(
732 queue_runner.add_queue_runner(queue_runner.QueueRunner(queue, enqueue_ops))
749 queue_runner.add_queue_runner(queue_runner.QueueRunner(queue, enqueue_ops))
Dsync_replicas_optimizer.py28 from tensorflow.python.training import queue_runner
335 self._chief_queue_runner = queue_runner.QueueRunner(
DBUILD53 "queue_runner.py",
99 ":queue_runner",
323 ":queue_runner",
459 name = "queue_runner",
460 srcs = ["queue_runner.py"],
527 ":queue_runner",
1497 ":queue_runner",
Dtraining.py47 from tensorflow.python.training.queue_runner import *
Dmonitored_session.py36 from tensorflow.python.training import queue_runner
910 queue_runner.start_queue_runners(sess=self.tf_sess, coord=self.coord)
Dsaver_test.py1906 queue_runner = queue_runner_pb2.QueueRunnerDef(queue_name="test_queue")
1908 str(queue_runner))
1910 queue_runner.SerializeToString())
1912 any_buf.Pack(queue_runner)
2800 queue_runner = queue_runner_pb2.QueueRunnerDef(queue_name="test_queue")
2802 str(queue_runner))
2804 queue_runner.SerializeToString())
2806 any_buf.Pack(queue_runner)
/external/tensorflow/tensorflow/cc/
DBUILD880 name = "queue_runner",
881 srcs = ["training/queue_runner.cc"],
882 hdrs = ["training/queue_runner.h"],
899 ":queue_runner",
930 ":queue_runner",
954 ":queue_runner",
/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"
Dtensorflow.train.pbtxt252 name: "queue_runner"
/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()
/external/tensorflow/tensorflow/core/protobuf/
DBUILD37 "queue_runner.proto",
/external/tensorflow/tensorflow/python/tools/api/generator/
Dapi_init_files_v1.bzl95 "train/queue_runner/__init__.py",
/external/tensorflow/
DAndroid.bp228 "tensorflow/core/protobuf/queue_runner.proto",
/external/tensorflow/tensorflow/core/
DBUILD189 "//tensorflow/core/protobuf:queue_runner.proto",