| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | fifo_queue_test.py | 70 enqueue_ops = [q.enqueue((x,)) for x in elems]
78 self.checkedThread(target=enqueue, args=(e,)) for e in enqueue_ops
95 enqueue_ops = [q.enqueue((x,)) for x in elems]
99 for enqueue_op in enqueue_ops:
108 threads = [self.checkedThread(target=dequeue) for _ in enqueue_ops]
119 enqueue_ops = [q.enqueue((x,)) for x in elems]
122 for enqueue_op in enqueue_ops:
133 enqueue_ops = [q.enqueue((x,)) for x in elems]
140 for enqueue_op in enqueue_ops:
163 enqueue_ops = [q.enqueue((x, y)) for x, y in elems]
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | queue_runner_impl.py | 58 def __init__(self, queue=None, enqueue_ops=None, close_op=None, argument
100 if queue or enqueue_ops:
106 queue=queue, enqueue_ops=enqueue_ops,
117 def _init_from_args(self, queue=None, enqueue_ops=None, close_op=None, argument
135 if not queue or not enqueue_ops:
138 self._enqueue_ops = enqueue_ops
196 def enqueue_ops(self): member in QueueRunner
370 for enqueue_op in self.enqueue_ops:
|
| H A D | input.py | 725 enqueue_ops = [enqueue_fn(_select_which_to_enqueue(x, keep_input))
728 enqueue_ops = [utils.smart_cond(
732 queue_runner.add_queue_runner(queue_runner.QueueRunner(queue, enqueue_ops))
742 enqueue_ops = [
745 enqueue_ops = [utils.smart_cond(
749 queue_runner.add_queue_runner(queue_runner.QueueRunner(queue, enqueue_ops))
|
| H A D | queue_runner_test.py | 329 self.assertEqual(1, len(qr0_recon.enqueue_ops))
330 self.assertEqual(enqueue_op, qr0_recon.enqueue_ops[0])
342 self.assertEqual(1, len(qr0_legacy_recon.enqueue_ops))
343 self.assertEqual(enqueue_op, qr0_legacy_recon.enqueue_ops[0])
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/data_structures/ |
| H A D | priority_queue_test.py | 79 enqueue_ops = [
92 target=enqueue, args=(op,)) for op in enqueue_ops
130 enqueue_ops = [
151 target=enqueue, args=(op,)) for op in enqueue_ops
183 enqueue_ops = [
207 target=enqueue, args=(op,)) for op in enqueue_ops
241 enqueue_ops = [
256 target=enqueue, args=(op,)) for op in enqueue_ops
|
| H A D | padding_fifo_queue_test.py | 124 enqueue_ops = [q.enqueue((x,)) for x in elems]
133 target=enqueue, args=(e,)) for e in enqueue_ops
153 enqueue_ops = [q.enqueue((x,)) for x in elems]
157 for enqueue_op in enqueue_ops:
166 threads = [self.checkedThread(target=dequeue) for _ in enqueue_ops]
177 enqueue_ops = [q.enqueue((x,)) for x in elems]
180 for enqueue_op in enqueue_ops:
194 enqueue_ops = [q.enqueue((x,)) for x in elems]
201 for enqueue_op in enqueue_ops:
226 enqueue_ops = [q.enqueue((x, y)) for x, y in elems]
[all …]
|
| H A D | fifo_queue_test.py | 777 enqueue_ops = [q.enqueue((x,)) for x in elems]
786 target=enqueue, args=(e,)) for e in enqueue_ops
806 enqueue_ops = [q.enqueue((x,)) for x in elems]
810 for enqueue_op in enqueue_ops:
819 threads = [self.checkedThread(target=dequeue) for _ in enqueue_ops]
833 enqueue_ops = [q.enqueue((x,)) for x in elems]
840 for enqueue_op in enqueue_ops:
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/random/ |
| H A D | random_shuffle_queue_test.py | 100 enqueue_ops = [q.enqueue((x,)) for x in elems]
109 target=enqueue, args=(e,)) for e in enqueue_ops
126 enqueue_ops = [q.enqueue((x,)) for x in elems]
130 for enqueue_op in enqueue_ops:
139 threads = [self.checkedThread(target=dequeue) for _ in enqueue_ops]
150 enqueue_ops = [q.enqueue((x,)) for x in elems]
153 for enqueue_op in enqueue_ops:
163 enqueue_ops = [q.enqueue((x,)) for x in elems]
170 for enqueue_op in enqueue_ops:
193 enqueue_ops = [q.enqueue((x, y)) for x, y in elems]
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.train.queue_runner.-queue-runner.pbtxt | 14 name: "enqueue_ops"
35 …argspec: "args=[\'self\', \'queue\', \'enqueue_ops\', \'close_op\', \'cancel_op\', \'queue_closed_…
|
| H A D | tensorflow.train.-queue-runner.pbtxt | 14 name: "enqueue_ops"
35 …argspec: "args=[\'self\', \'queue\', \'enqueue_ops\', \'close_op\', \'cancel_op\', \'queue_closed_…
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/ |
| H A D | grappler_item.cc | 77 std::vector<string> enqueue_ops; in EnqueueOpsFanin() local
80 enqueue_ops.push_back(enqueue_op); in EnqueueOpsFanin()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | tpu_feed.py | 860 enqueue_ops = []
896 enqueue_ops.append(
903 return enqueue_ops
|
| H A D | tpu_embedding_v2.py | 1294 enqueue_ops = []
1314 enqueue_ops.append(enqueue_op)
1315 ops.get_default_graph().control_outputs.extend(enqueue_ops)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/grappler/ |
| H A D | item_wrapper.cc | 160 std::vector<const tensorflow::NodeDef*> enqueue_ops = in PYBIND11_MODULE() local
166 for (auto op : enqueue_ops) { in PYBIND11_MODULE()
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/training/ |
| H A D | queue_runner_test.cc | 84 const std::string& queue_name, const std::vector<std::string>& enqueue_ops, in BuildQueueRunnerDef() argument
89 for (const std::string& enqueue_op : enqueue_ops) { in BuildQueueRunnerDef()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/ |
| H A D | metrics_test.py | 1394 def _enqueue_as_batches(x, enqueue_ops): argument
1399 enqueue_ops[i].append(x_queue.enqueue(x_batches[i, :]))
1407 enqueue_ops = [[] for i in range(num_batches)]
1408 tf_predictions = _enqueue_as_batches(predictions, enqueue_ops)
1409 tf_labels = _enqueue_as_batches(labels, enqueue_ops)
1410 tf_weights = (_enqueue_as_batches(weights, enqueue_ops) if
1414 sess.run(enqueue_ops[i])
|