code: def _return_args_from_infinite_loop(
      *loop_vars, additional_loop_vars=()
  ):
      return tf.while_loop(
          cond=lambda *_: True,
          body=lambda *loop_vars: loop_vars,
          loop_vars=loop_vars + additional_loop_vars,
      )
  
file_path: probability/tensorflow_probability/python/internal/callable_util_test.py
imports: import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import callable_util
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 1 for calling a global function with example usage: _return_args_from_infinite_loop() and returns: tf