Function loop_fn

tensorflow/python/ops/parallel_for/gradients.py:53–55

(i)

  output = array_ops.reshape(output, [-1])

  def loop_fn(i):
    y = array_ops.gather(output, i)
    return gradient_ops.gradients(y, flat_inputs)

  try:
    output_size = int(output.shape[0])