| /external/tensorflow/tensorflow/python/estimator/inputs/queues/ |
| feeding_queue_runner.py | 32 def __init__(self, queue=None, enqueue_ops=None, close_op=None,
43 enqueue_ops: List of enqueue ops to run in threads later.
47 `Tensor`s to values. Must be the same length as `enqueue_ops`.
55 `enqueue_ops`.
61 queue, enqueue_ops, close_op,
64 self._feed_fns = [None for _ in enqueue_ops]
66 if len(feed_fns) != len(enqueue_ops):
69 "enqueue_ops.")
127 op passed in `enqueue_ops`.
|
| feeding_functions.py | 465 enqueue_ops = []
473 enqueue_ops.append(queue.enqueue_many(placeholders))
498 enqueue_ops=enqueue_ops,
|
| /external/tensorflow/tensorflow/python/training/ |
| queue_runner_impl.py | 56 def __init__(self, queue=None, enqueue_ops=None, close_op=None,
65 create one thread for each op in `enqueue_ops`. Each thread will run its
72 enqueue_ops: List of enqueue ops to run in threads later.
88 ValueError: If `queue` or `enqueue_ops` are not provided when not
98 if queue or enqueue_ops:
104 queue=queue, enqueue_ops=enqueue_ops,
115 def _init_from_args(self, queue=None, enqueue_ops=None, close_op=None,
121 enqueue_ops: List of enqueue ops to run in threads later.
128 ValueError: If `queue` or `enqueue_ops` are not provided when no
194 def enqueue_ops(self): member in class:QueueRunner
[all...] |
| queue_runner_test.py | 331 self.assertEqual(1, len(qr0_recon.enqueue_ops))
332 self.assertEqual(enqueue_op, qr0_recon.enqueue_ops[0])
344 self.assertEqual(1, len(qr0_legacy_recon.enqueue_ops))
345 self.assertEqual(enqueue_op, qr0_legacy_recon.enqueue_ops[0])
|
| input.py | 703 enqueue_ops = [enqueue_fn(_select_which_to_enqueue(x, keep_input))
706 enqueue_ops = [utils.smart_cond(
710 queue_runner.add_queue_runner(queue_runner.QueueRunner(queue, enqueue_ops))
720 enqueue_ops = [
723 enqueue_ops = [utils.smart_cond(
727 queue_runner.add_queue_runner(queue_runner.QueueRunner(queue, enqueue_ops))
[all...] |
| /external/tensorflow/tensorflow/core/grappler/ |
| grappler_item.cc | 48 std::vector<string> enqueue_ops; local
51 enqueue_ops.push_back(enqueue_op);
54 return ComputeTransitiveFanin(graph, enqueue_ops);
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| priority_queue_test.py | 77 enqueue_ops = [
90 target=enqueue, args=(op,)) for op in enqueue_ops
125 enqueue_ops = [
146 target=enqueue, args=(op,)) for op in enqueue_ops
175 enqueue_ops = [
199 target=enqueue, args=(op,)) for op in enqueue_ops
230 enqueue_ops = [
245 target=enqueue, args=(op,)) for op in enqueue_ops
|
| 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
150 enqueue_ops = [q.enqueue((x,)) for x in elems]
154 for enqueue_op in enqueue_ops:
163 threads = [self.checkedThread(target=dequeue) for _ in enqueue_ops]
174 enqueue_ops = [q.enqueue((x,)) for x in elems]
177 for enqueue_op in enqueue_ops:
188 enqueue_ops = [q.enqueue((x,)) for x in elems]
192 # The enqueue_ops should run after the dequeue op has blocked.
195 for enqueue_op in enqueue_ops
[all...] |
| fifo_queue_test.py | 147 enqueue_ops = [q.enqueue((x,)) for x in elems]
156 target=enqueue, args=(e,)) for e in enqueue_ops
173 enqueue_ops = [q.enqueue((x,)) for x in elems]
177 for enqueue_op in enqueue_ops:
186 threads = [self.checkedThread(target=dequeue) for _ in enqueue_ops]
197 enqueue_ops = [q.enqueue((x,)) for x in elems]
200 for enqueue_op in enqueue_ops:
211 enqueue_ops = [q.enqueue((x,)) for x in elems]
214 for enqueue_op in enqueue_ops:
225 enqueue_ops = [q.enqueue((x,)) for x in elems
[all...] |
| metrics_test.py | [all...] |
| /external/tensorflow/tensorflow/contrib/slim/python/slim/data/ |
| parallel_reader.py | 160 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/contrib/tpu/python/tpu/ |
| tpu_estimator.py | 320 enqueue_ops,
324 self._enqueue_ops = enqueue_ops
487 def __init__(self, ctx, enqueue_ops, dequeue_ops):
489 ctx, enqueue_ops, dequeue_ops, run_infeed_loop_on_coordinator=False)
629 """A fn returns enqueue_ops."""
[all...] |
| /external/tensorflow/tensorflow/python/grappler/ |
| item.i | 106 std::vector<const tensorflow::NodeDef*> enqueue_ops = item->EnqueueOpsFanin();
111 for (auto op : enqueue_ops) {
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| random_shuffle_queue_test.py | 99 enqueue_ops = [q.enqueue((x,)) for x in elems]
108 target=enqueue, args=(e,)) for e in enqueue_ops
125 enqueue_ops = [q.enqueue((x,)) for x in elems]
129 for enqueue_op in enqueue_ops:
138 threads = [self.checkedThread(target=dequeue) for _ in enqueue_ops]
149 enqueue_ops = [q.enqueue((x,)) for x in elems]
152 for enqueue_op in enqueue_ops:
162 enqueue_ops = [q.enqueue((x,)) for x in elems]
166 # The enqueue_ops should run after the dequeue op has blocked.
169 for enqueue_op in enqueue_ops
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/ |
| graph_io.py | 702 enqueue_ops = [
709 enqueue_ops,
[all...] |
| graph_io_test.py | 567 for op in qr.enqueue_ops:
|
| /external/tensorflow/tensorflow/cc/training/ |
| queue_runner_test.cc | 84 const std::string& queue_name, const std::vector<std::string>& enqueue_ops,
89 for (const std::string& enqueue_op : enqueue_ops) {
|
| /external/tensorflow/tensorflow/contrib/metrics/python/ops/ |
| metric_ops_test.py | [all...] |
