Lines Matching defs:ops
21 from tensorflow.python.data.ops import dataset_ops
22 from tensorflow.python.framework import ops as tf_ops
28 Constructs an infinitely repeating sequence of Ops in shuffled order.
34 def __init__(self, ops, seed=None):
38 ops: list of TensorFlow Ops. Ops to be selected from. All workers must
39 initialize with the same set of ops.
40 seed: int or None. Random seed used when shuffling order of ops.
42 self._ops_by_name = {op.name: op for op in ops}
45 op_names = tf_ops.convert_to_tensor(list(sorted(op.name for op in ops)))
47 .shuffle(len(ops), seed=seed).repeat())
51 def ops(self):
52 """Ops this OpQueue can return in next_op()."""
64 Next Op chosen from 'ops'.
