Home
last modified time | relevance | path

Searched refs:enqueue_ops (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/tests/
Dfifo_queue_test.py76 enqueue_ops = [q.enqueue((x,)) for x in elems]
84 self.checkedThread(target=enqueue, args=(e,)) for e in enqueue_ops
101 enqueue_ops = [q.enqueue((x,)) for x in elems]
105 for enqueue_op in enqueue_ops:
114 threads = [self.checkedThread(target=dequeue) for _ in enqueue_ops]
125 enqueue_ops = [q.enqueue((x,)) for x in elems]
128 for enqueue_op in enqueue_ops:
139 enqueue_ops = [q.enqueue((x,)) for x in elems]
146 for enqueue_op in enqueue_ops:
169 enqueue_ops = [q.enqueue((x, y)) for x, y in elems]
[all …]
/external/tensorflow/tensorflow/python/training/
Dqueue_runner_impl.py61 def __init__(self, queue=None, enqueue_ops=None, close_op=None, argument
103 if queue or enqueue_ops:
109 queue=queue, enqueue_ops=enqueue_ops,
120 def _init_from_args(self, queue=None, enqueue_ops=None, close_op=None, argument
138 if not queue or not enqueue_ops:
141 self._enqueue_ops = enqueue_ops
199 def enqueue_ops(self): member in QueueRunner
373 for enqueue_op in self.enqueue_ops:
Dqueue_runner_test.py333 self.assertEqual(1, len(qr0_recon.enqueue_ops))
334 self.assertEqual(enqueue_op, qr0_recon.enqueue_ops[0])
346 self.assertEqual(1, len(qr0_legacy_recon.enqueue_ops))
347 self.assertEqual(enqueue_op, qr0_legacy_recon.enqueue_ops[0])
Dinput.py728 enqueue_ops = [enqueue_fn(_select_which_to_enqueue(x, keep_input))
731 enqueue_ops = [utils.smart_cond(
735 queue_runner.add_queue_runner(queue_runner.QueueRunner(queue, enqueue_ops))
745 enqueue_ops = [
748 enqueue_ops = [utils.smart_cond(
752 queue_runner.add_queue_runner(queue_runner.QueueRunner(queue, enqueue_ops))
/external/tensorflow/tensorflow/python/kernel_tests/
Dpriority_queue_test.py83 enqueue_ops = [
96 target=enqueue, args=(op,)) for op in enqueue_ops
134 enqueue_ops = [
155 target=enqueue, args=(op,)) for op in enqueue_ops
187 enqueue_ops = [
211 target=enqueue, args=(op,)) for op in enqueue_ops
245 enqueue_ops = [
260 target=enqueue, args=(op,)) for op in enqueue_ops
Dpadding_fifo_queue_test.py129 enqueue_ops = [q.enqueue((x,)) for x in elems]
138 target=enqueue, args=(e,)) for e in enqueue_ops
158 enqueue_ops = [q.enqueue((x,)) for x in elems]
162 for enqueue_op in enqueue_ops:
171 threads = [self.checkedThread(target=dequeue) for _ in enqueue_ops]
182 enqueue_ops = [q.enqueue((x,)) for x in elems]
185 for enqueue_op in enqueue_ops:
199 enqueue_ops = [q.enqueue((x,)) for x in elems]
206 for enqueue_op in enqueue_ops:
231 enqueue_ops = [q.enqueue((x, y)) for x, y in elems]
[all …]
Dfifo_queue_test.py782 enqueue_ops = [q.enqueue((x,)) for x in elems]
791 target=enqueue, args=(e,)) for e in enqueue_ops
811 enqueue_ops = [q.enqueue((x,)) for x in elems]
815 for enqueue_op in enqueue_ops:
824 threads = [self.checkedThread(target=dequeue) for _ in enqueue_ops]
838 enqueue_ops = [q.enqueue((x,)) for x in elems]
845 for enqueue_op in enqueue_ops:
Dmetrics_test.py1399 def _enqueue_as_batches(x, enqueue_ops): argument
1404 enqueue_ops[i].append(x_queue.enqueue(x_batches[i, :]))
1412 enqueue_ops = [[] for i in range(num_batches)]
1413 tf_predictions = _enqueue_as_batches(predictions, enqueue_ops)
1414 tf_labels = _enqueue_as_batches(labels, enqueue_ops)
1415 tf_weights = (_enqueue_as_batches(weights, enqueue_ops) if
1419 sess.run(enqueue_ops[i])
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.queue_runner.-queue-runner.pbtxt14 name: "enqueue_ops"
35 …argspec: "args=[\'self\', \'queue\', \'enqueue_ops\', \'close_op\', \'cancel_op\', \'queue_closed_…
Dtensorflow.train.-queue-runner.pbtxt14 name: "enqueue_ops"
35 …argspec: "args=[\'self\', \'queue\', \'enqueue_ops\', \'close_op\', \'cancel_op\', \'queue_closed_…
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item.cc77 std::vector<string> enqueue_ops; in EnqueueOpsFanin() local
80 enqueue_ops.push_back(enqueue_op); in EnqueueOpsFanin()
/external/tensorflow/tensorflow/python/kernel_tests/random/
Drandom_shuffle_queue_test.py105 enqueue_ops = [q.enqueue((x,)) for x in elems]
114 target=enqueue, args=(e,)) for e in enqueue_ops
131 enqueue_ops = [q.enqueue((x,)) for x in elems]
135 for enqueue_op in enqueue_ops:
144 threads = [self.checkedThread(target=dequeue) for _ in enqueue_ops]
155 enqueue_ops = [q.enqueue((x,)) for x in elems]
158 for enqueue_op in enqueue_ops:
168 enqueue_ops = [q.enqueue((x,)) for x in elems]
175 for enqueue_op in enqueue_ops:
198 enqueue_ops = [q.enqueue((x, y)) for x, y in elems]
[all …]
/external/tensorflow/tensorflow/python/grappler/
Ditem_wrapper.cc160 std::vector<const tensorflow::NodeDef*> enqueue_ops = in PYBIND11_MODULE() local
166 for (auto op : enqueue_ops) { in PYBIND11_MODULE()
/external/tensorflow/tensorflow/python/tpu/
Dtpu_feed.py859 enqueue_ops = []
895 enqueue_ops.append(
902 return enqueue_ops
Dtpu_embedding_v2.py1298 enqueue_ops = []
1317 enqueue_ops.append(enqueue_op)
1318 ops.get_default_graph().control_outputs.extend(enqueue_ops)
/external/tensorflow/tensorflow/cc/training/
Dqueue_runner_test.cc84 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()