Home
last modified time | relevance | path

Searched refs:enqueue_op (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dfifo_queue_test.py105 enqueue_op = q.enqueue((10.0,))
106 enqueue_op.run()
111 enqueue_op = q.enqueue((10.0,))
112 enqueue_op.run()
162 def enqueue(enqueue_op): argument
163 self.evaluate(enqueue_op)
188 for enqueue_op in enqueue_ops:
189 enqueue_op.run()
211 for enqueue_op in enqueue_ops:
212 enqueue_op.run()
[all …]
Dpadding_fifo_queue_test.py102 enqueue_op = q.enqueue((10.0,))
103 enqueue_op.run()
130 def enqueue(enqueue_op): argument
131 self.evaluate(enqueue_op)
156 for enqueue_op in enqueue_ops:
157 enqueue_op.run()
179 for enqueue_op in enqueue_ops:
180 enqueue_op.run()
197 for enqueue_op in enqueue_ops:
198 self.evaluate(enqueue_op)
[all …]
Dpriority_queue_test.py85 def enqueue(enqueue_op): argument
86 self.evaluate(enqueue_op)
136 def enqueue(enqueue_op): argument
137 self.evaluate(enqueue_op)
188 def enqueue(enqueue_op): argument
189 self.evaluate(enqueue_op)
240 def enqueue(enqueue_op): argument
241 self.evaluate(enqueue_op)
/external/tensorflow/tensorflow/python/kernel_tests/random/
Drandom_shuffle_queue_test.py53 enqueue_op = q.enqueue((10.0,))
55 enqueue_op.run()
105 def enqueue(enqueue_op): argument
106 self.evaluate(enqueue_op)
131 for enqueue_op in enqueue_ops:
132 enqueue_op.run()
154 for enqueue_op in enqueue_ops:
155 enqueue_op.run()
171 for enqueue_op in enqueue_ops:
172 self.evaluate(enqueue_op)
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Dfifo_queue_test.py36 enqueue_op = q.enqueue((10.0,))
37 enqueue_op.run()
80 def enqueue(enqueue_op): argument
81 sess.run(enqueue_op)
105 for enqueue_op in enqueue_ops:
106 enqueue_op.run()
128 for enqueue_op in enqueue_ops:
129 enqueue_op.run()
146 for enqueue_op in enqueue_ops:
147 sess.run(enqueue_op)
[all …]
/external/tensorflow/tensorflow/contrib/gan/python/features/python/
Drandom_tensor_pool_impl.py112 enqueue_op = pool_queue.enqueue(input_values)
113 with ops.control_dependencies([enqueue_op]):
119 enqueue_op = pool_queue.enqueue(input_values)
120 with ops.control_dependencies([enqueue_op]):
/external/tensorflow/tensorflow/compiler/xla/tests/
Dexhaustive_op_test.cc197 void Run(std::function<XlaOp(XlaOp)> enqueue_op, in Run() argument
211 RunImpl<float, uint32>(enqueue_op, evaluate_op); in Run()
215 RunImpl<half, uint16>(enqueue_op, evaluate_op); in Run()
219 RunImpl<bfloat16, uint16>(enqueue_op, evaluate_op); in Run()
236 void RunImpl(std::function<XlaOp(XlaOp)> enqueue_op, in RunImpl() argument
274 BuildAndRunComputation(enqueue_op, input_literal)); in RunImpl()
279 const std::function<XlaOp(XlaOp)>& enqueue_op, in BuildAndRunComputation() argument
283 enqueue_op(input); in BuildAndRunComputation()
/external/tensorflow/tensorflow/cc/training/
Dqueue_runner.cc98 for (const string& enqueue_op : enqueue_op_names_) { in Start() local
100 std::bind(&QueueRunner::Run, this, sess, enqueue_op)); in Start()
162 void QueueRunner::Run(Session* sess, const string& enqueue_op) { in Run() argument
169 status = RealRun(sess, enqueue_op, true); in Run()
Dqueue_runner.h95 void Run(Session* sess, const string& enqueue_op);
Dqueue_runner_test.cc89 for (const std::string& enqueue_op : enqueue_ops) { in BuildQueueRunnerDef() local
90 *queue_runner_def.mutable_enqueue_op_name()->Add() = enqueue_op; in BuildQueueRunnerDef()
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item.cc58 for (const string& enqueue_op : queue_runner.enqueue_op_name()) { in EnqueueOpsFanin() local
59 enqueue_ops.push_back(enqueue_op); in EnqueueOpsFanin()
107 for (const string& enqueue_op : queue_runner.enqueue_op_name()) { in NodesToPreserve() local
108 result.insert(NodeName(enqueue_op)); in NodesToPreserve()
/external/tensorflow/tensorflow/contrib/slim/python/slim/data/
Dprefetch_queue.py87 enqueue_op = queue.enqueue(tensors)
89 queue_runner.QueueRunner(queue, [enqueue_op] * num_threads))
/external/tensorflow/tensorflow/python/training/
Dqueue_runner_impl.py239 def _run(self, sess, enqueue_op, coord=None): argument
252 enqueue_callable = sess.make_callable(enqueue_op)
373 for enqueue_op in self.enqueue_ops:
375 ops.strip_name_scope(enqueue_op.name, export_scope))
Dqueue_runner_test.py321 enqueue_op = control_flow_ops.no_op(name="enqueue")
325 queue, [enqueue_op],
334 self.assertEqual(enqueue_op, qr0_recon.enqueue_ops[0])
347 self.assertEqual(enqueue_op, qr0_legacy_recon.enqueue_ops[0])
Dserver_lib_test.py158 enqueue_op = q.enqueue(37.0)
161 sess.run(enqueue_op)
/external/tensorflow/tensorflow/contrib/training/python/training/
Dsampling_ops_threading_test.py56 enqueue_op = queue.enqueue([prob_estimate])
58 queue_runner_impl.QueueRunner(queue, [enqueue_op] * 25))
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dgmm_test.py206 enqueue_op = queue.enqueue(array_ops.zeros([10, 3], dtype=dtypes.float32))
208 [enqueue_op]))
Dkmeans_test.py599 enqueue_op = queue.enqueue(array_ops.zeros([10, 3], dtype=dtypes.float32))
601 queue_runner.QueueRunner(queue, [enqueue_op]))
/external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/
Dgraph_io.py409 enqueue_op = file_name_queue.enqueue(
416 queue_runner.QueueRunner(file_name_queue, [enqueue_op]))
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dmap_test.py361 enqueue_op = queue.enqueue_many(elements)
367 self.evaluate(enqueue_op)
384 enqueue_op = queue.enqueue_many(elements)
390 self.evaluate(enqueue_op)
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dkmeans_test.py568 enqueue_op = queue.enqueue(array_ops.zeros([10, 3], dtype=dtypes.float32))
570 queue_runner.QueueRunner(queue, [enqueue_op]))
/external/tensorflow/tensorflow/python/tpu/
Dtpu_estimator.py2831 for enqueue_op in enqueue_ops:
2832 if isinstance(enqueue_op, list):
2833 graph.get_collection_ref(_TPU_ENQUEUE_OPS).extend(enqueue_op)
2835 graph.add_to_collection(_TPU_ENQUEUE_OPS, enqueue_op)
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc8499 TF_Operation* enqueue_op = TF_GraphOperationByName( in TF_EnqueueNamedTensor() local
8502 if (enqueue_op == nullptr) { in TF_EnqueueNamedTensor()
8525 /*targets*/ &enqueue_op, /*ntargets*/ 1, in TF_EnqueueNamedTensor()
/external/tensorflow/tensorflow/python/client/
Dsession_test.py1842 enqueue_op = q.enqueue_many(constant_op.constant([1, 2]))
1849 sess.run(enqueue_op)