Lines Matching refs:ops
21 from tensorflow.contrib.kfac.python.ops import op_queue
22 from tensorflow.python.framework import ops as tf_ops
23 from tensorflow.python.ops import math_ops
30 """Ensures all ops get selected eventually."""
32 ops = [
37 queue = op_queue.OpQueue(ops, seed=0)
41 selected_ops = set([queue.next_op(sess) for _ in ops])
42 self.assertEqual(set(ops), set(selected_ops))
44 # Ensure additional calls don't create any new ops.
46 self.assertEqual(set(ops), set(selected_ops))
