code: def test_stateless_while_2var():
      class StatelessWhile2Var(tf.Module):
          def get_input(self):
              return np.array([20], dtype="float32")
  
          @tf.function(
              input_signature=[
                  tf.TensorSpec(shape=(1,), dtype=tf.float32)
              ]
          )
          def func(self, x):
              i = tf.constant(3.0)
              j = tf.constant(5.0)
              cond = lambda i, j: tf.less(i + j, x)
              body = lambda i, j: (tf.add(i, 2), tf.add(j, 3))
              r = tf.while_loop(cond, body, [i, j])
              return r
  
      run_func_graph(
          StatelessWhile2Var,
          runtime="vm",
          outputs=["Identity:output:0", "Identity_1:output:0"],
      )
      run_model_graph(StatelessWhile2Var, outputs=["Identity:output:0"])
  
imports: import tempfile
  import tensorflow
  import numpy
  import pytest
  from common import compare_tf_tvm
  from common import run_tf_code
label_desc: Use image node_id 32 for calling a global function with example usage: test_stateless_while_2var() and returns: np, r