Method fnWithLoop

tensorflow/python/kernel_tests/while_v2_test.py:119–125

()

      @def_function.function
      def fnWithLoop():  # pylint: disable=invalid-name
        with backprop.GradientTape() as tape:
          _, x = while_loop_v2(
              lambda i, _: i < 2,
              lambda i, x: (i + 1, x * v),
              [0, 2.])
        return tape.gradient(x, v)

      self.assertAllEqual(fnWithLoop(), 4.0)