code: def test_while_loop_power(self, compute_unit, backend):
      @make_tf_graph([(1,)])
      def build_model(x):
          i = 0
          while i < 3:
              x *= x
              i += 1
          return x
  
      model, inputs, outputs = build_model
      input_values = [np.array([2.0], dtype=np.float32)]
      input_dict = dict(zip(inputs, input_values))
      TensorFlowBaseTest.run_compare_tf(
          model,
          input_dict,
          outputs,
          compute_unit=compute_unit,
          backend=backend,
      )
  
imports: import itertools
  import numpy
  import pytest
  import coremltools
  from coremltools.converters.mil import testing_reqs
  from coremltools.converters.mil.frontend.tensorflow.test.testing_utils import TensorFlowBaseTest
  from coremltools.converters.mil.frontend.tensorflow2.test.testing_utils import TensorFlow2BaseTest
  from coremltools.converters.mil.frontend.tensorflow2.test.testing_utils import make_tf2_graph
  from coremltools.converters.mil.testing_utils import random_gen
label_desc: Use image node_id 5 for calling the TestControlFlowFromAutoGraph obj's underlying member method code with example usage: obj.test_while_loop_power(compute_unit, backend) and returns: x