Home
last modified time | relevance | path

Searched refs:enqueue_ops (Results 1 – 19 of 19) 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.py729 enqueue_ops = [enqueue_fn(_select_which_to_enqueue(x, keep_input))
732 enqueue_ops = [utils.smart_cond(
736 queue_runner.add_queue_runner(queue_runner.QueueRunner(queue, enqueue_ops))
746 enqueue_ops = [
749 enqueue_ops = [utils.smart_cond(
753 queue_runner.add_queue_runner(queue_runner.QueueRunner(queue, enqueue_ops))
/external/tensorflow/tensorflow/python/kernel_tests/
Dpriority_queue_test.py79 enqueue_ops = [
92 target=enqueue, args=(op,)) for op in enqueue_ops
127 enqueue_ops = [
148 target=enqueue, args=(op,)) for op in enqueue_ops
177 enqueue_ops = [
201 target=enqueue, args=(op,)) for op in enqueue_ops
232 enqueue_ops = [
247 target=enqueue, args=(op,)) for op in enqueue_ops
Dpadding_fifo_queue_test.py126 enqueue_ops = [q.enqueue((x,)) for x in elems]
135 target=enqueue, args=(e,)) for e in enqueue_ops
152 enqueue_ops = [q.enqueue((x,)) for x in elems]
156 for enqueue_op in enqueue_ops:
165 threads = [self.checkedThread(target=dequeue) for _ in enqueue_ops]
176 enqueue_ops = [q.enqueue((x,)) for x in elems]
179 for enqueue_op in enqueue_ops:
190 enqueue_ops = [q.enqueue((x,)) for x in elems]
197 for enqueue_op in enqueue_ops:
222 enqueue_ops = [q.enqueue((x, y)) for x, y in elems]
[all …]
Dfifo_queue_test.py158 enqueue_ops = [q.enqueue((x,)) for x in elems]
167 target=enqueue, args=(e,)) for e in enqueue_ops
184 enqueue_ops = [q.enqueue((x,)) for x in elems]
188 for enqueue_op in enqueue_ops:
197 threads = [self.checkedThread(target=dequeue) for _ in enqueue_ops]
208 enqueue_ops = [q.enqueue((x,)) for x in elems]
211 for enqueue_op in enqueue_ops:
222 enqueue_ops = [q.enqueue((x,)) for x in elems]
225 for enqueue_op in enqueue_ops:
236 enqueue_ops = [q.enqueue((x,)) for x in elems]
[all …]
Dmetrics_test.py1371 def _enqueue_as_batches(x, enqueue_ops): argument
1376 enqueue_ops[i].append(x_queue.enqueue(x_batches[i, :]))
1384 enqueue_ops = [[] for i in range(num_batches)]
1385 tf_predictions = _enqueue_as_batches(predictions, enqueue_ops)
1386 tf_labels = _enqueue_as_batches(labels, enqueue_ops)
1387 tf_weights = (_enqueue_as_batches(weights, enqueue_ops) if
1391 sess.run(enqueue_ops[i])
/external/tensorflow/tensorflow/contrib/slim/python/slim/data/
Dparallel_reader.py160 enqueue_ops = []
162 enqueue_ops.append(self._common_queue.enqueue(reader.read(queue)))
165 queue_runner.QueueRunner(self._common_queue, enqueue_ops))
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.-queue-runner.pbtxt14 name: "enqueue_ops"
35 …argspec: "args=[\'self\', \'queue\', \'enqueue_ops\', \'close_op\', \'cancel_op\', \'queue_closed_…
Dtensorflow.train.queue_runner.-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.cc56 std::vector<string> enqueue_ops; in EnqueueOpsFanin() local
59 enqueue_ops.push_back(enqueue_op); in EnqueueOpsFanin()
62 return ComputeTransitiveFanin(graph, enqueue_ops); in EnqueueOpsFanin()
/external/tensorflow/tensorflow/python/tpu/
Dtpu_estimator.py439 enqueue_ops, argument
448 self._enqueue_ops = enqueue_ops
587 def __init__(self, ctx, enqueue_ops, dequeue_ops, tpu_compile_op, argument
591 enqueue_ops,
1021 enqueue_ops = infeed_queue.generate_enqueue_ops(
1027 return enqueue_ops
1030 'ops': enqueue_ops,
1217 enqueue_ops, all_hooks, run_infeed_loop_on_coordinator = (
1232 return (enqueue_ops, dequeue_fn, all_hooks, run_infeed_loop_on_coordinator)
1236 enqueue_ops = []
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/random/
Drandom_shuffle_queue_test.py101 enqueue_ops = [q.enqueue((x,)) for x in elems]
110 target=enqueue, args=(e,)) for e in enqueue_ops
127 enqueue_ops = [q.enqueue((x,)) for x in elems]
131 for enqueue_op in enqueue_ops:
140 threads = [self.checkedThread(target=dequeue) for _ in enqueue_ops]
151 enqueue_ops = [q.enqueue((x,)) for x in elems]
154 for enqueue_op in enqueue_ops:
164 enqueue_ops = [q.enqueue((x,)) for x in elems]
171 for enqueue_op in enqueue_ops:
194 enqueue_ops = [q.enqueue((x, y)) for x, y in elems]
[all …]
/external/tensorflow/tensorflow/python/grappler/
Ditem.i105 std::vector<const tensorflow::NodeDef*> enqueue_ops = item->EnqueueOpsFanin(); in TF_IdentifyImportantOps() local
110 for (auto op : enqueue_ops) { in TF_IdentifyImportantOps()
/external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/
Dgraph_io.py714 enqueue_ops = [
721 enqueue_ops,
Dgraph_io_test.py567 for op in qr.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()
/external/tensorflow/tensorflow/contrib/metrics/python/ops/
Dmetric_ops_test.py1884 def _enqueue_as_batches(x, enqueue_ops): argument
1889 enqueue_ops[i].append(x_queue.enqueue(x_batches[i, :]))
1897 enqueue_ops = [[] for i in range(num_batches)]
1898 tf_predictions = _enqueue_as_batches(predictions, enqueue_ops)
1899 tf_labels = _enqueue_as_batches(labels, enqueue_ops)
1901 _enqueue_as_batches(weights, enqueue_ops)
1905 sess.run(enqueue_ops[i])