code: def _make_iteration_function(encoder):
      assert isinstance(encoder, simple_encoder.SimpleEncoder)
  
      @tf.function
      def iteration(x, state):
          encoded_x, new_state = encoder.encode(x, state)
          decoded_x = encoder.decode(encoded_x)
          return x, encoded_x, decoded_x, new_state
  
      return iteration
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import tensorflow
  from tensorflow.python.framework import test_util
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.core import core_encoder
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.core import simple_encoder
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.testing import test_utils
label_desc: Use image node_id 1 for calling a global function with example usage: _make_iteration_function(encoder) and returns: iteration, x,  encoded_x,  decoded_x,  new_state