(device, op_fn)
Wraps the ops generated by `op_fn` in tf.while_loop.
Wraps the ops generated by `op_fn` in tf.while_loop.
Wraps the ops generated by op_fn in tf . while_loop .
def _wrap_computation_in_while_loop(device, op_fn):
  """Wraps the ops generated by `op_fn` in tf.while_loop."""

  def computation(i):
    with ops.control_dependencies(op_fn()):
      return i + 1

  iterations_per_loop_var = _create_or_get_iterations_per_loop()
  # By setting parallel_iterations=1, the parallel execution in while_loop is
  # basically turned off.
  with ops.device(device):
    iterations = array_ops.identity(iterations_per_loop_var)
    return control_flow_ops.while_loop(
        lambda i: i < iterations,
        computation, [constant_op.constant(0)],
        parallel_iterations=1)
def _wrap_computation_in_while_loop ( device , op_fn ) : def computation ( i ) : with ops . control_dependencies ( op_fn ( ) ) : return i + 1 iterations_per_loop_var = _create_or_get_iterations_per_loop ( ) # By setting parallel_iterations=1, the parallel execution in while_loop is # basically turned off. with ops . device ( device ) : iterations = array_ops . identity ( iterations_per_loop_var ) return control_flow_ops . while_loop ( lambda i : i < iterations , computation , [ constant_op . constant ( 0 ) ] , parallel_iterations = 1 )