| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/ |
| H A D | grappler_item.cc | 78 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()
|
| H A D | grappler_item_builder.cc | 526 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/inputs/ |
| H A D | trivial_test_graph_input_yielder.cc | 113 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/clusters/ |
| H A D | single_machine.cc | 175 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()
|
| H A D | single_machine_test.cc | 541 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()
|
| H A D | BUILD | 130 "//tensorflow/cc:queue_runner",
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | input.py | 40 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))
|
| H A D | BUILD | 53 "queue_runner.py",
99 ":queue_runner",
323 ":queue_runner",
459 name = "queue_runner",
460 srcs = ["queue_runner.py"],
527 ":queue_runner",
1497 ":queue_runner",
|
| H A D | sync_replicas_optimizer.py | 28 from tensorflow.python.training import queue_runner
335 self._chief_queue_runner = queue_runner.QueueRunner(
|
| H A D | saver_test.py | 1906 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)
|
| H A D | training.py | 47 from tensorflow.python.training.queue_runner import *
|
| H A D | monitored_session.py | 36 from tensorflow.python.training import queue_runner
910 queue_runner.start_queue_runners(sess=self.tf_sess, coord=self.coord)
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/ |
| H A D | BUILD | 880 name = "queue_runner",
881 srcs = ["training/queue_runner.cc"],
882 hdrs = ["training/queue_runner.h"],
899 ":queue_runner",
930 ":queue_runner",
954 ":queue_runner",
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/ |
| D | queue_runner.pb.d | 1 …_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/queue_runner.pb.cc: external…
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.train.queue_runner.pbtxt | 1 path: "tensorflow.train.queue_runner"
|
| H A D | tensorflow.train.queue_runner.-queue-runner.pbtxt | 1 path: "tensorflow.train.queue_runner.QueueRunner"
|
| H A D | tensorflow.train.pbtxt | 252 name: "queue_runner"
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/training/ |
| H A D | queue_runner_test.cc | 231 void JoinThread(QueueRunner* queue_runner, bool* join_succeeded, in JoinThread() argument
233 EXPECT_EQ(queue_runner->Join().code(), Code::CANCELLED); in JoinThread()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/ |
| H A D | BUILD | 37 "queue_runner.proto",
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_java_lite/android_common_apex33/ |
| D | meta_lic.rsp | 1 …uf/named_tensor.proto -s external/tensorflow/tensorflow/core/protobuf/queue_runner.proto -s extern…
|
| D | meta_lic | 47 sources: "external/tensorflow/tensorflow/core/protobuf/queue_runner.proto"
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tools/api/generator/ |
| H A D | api_init_files_v1.bzl | 95 "train/queue_runner/__init__.py",
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/ |
| D | meta_lic.rsp | 1 …uf/named_tensor.proto -s external/tensorflow/tensorflow/core/protobuf/queue_runner.proto -s extern…
|
| D | meta_lic | 47 sources: "external/tensorflow/tensorflow/core/protobuf/queue_runner.proto"
|
| /aosp_15_r20/external/tensorflow/ |
| H A D | Android.bp | 261 "tensorflow/core/protobuf/queue_runner.proto",
|