code: def _get_serve_tf_examples_fn(model, tf_transform_output):
      """Returns a function that feeds the input tensor into the model."""
  
      model.tft_layer = tf_transform_output.transform_features_layer()
  
      @tf.function
      def serve_tf_examples_fn(image_tensor):
          """Returns the output to be used in the serving signature."""
          transformed_features = model.tft_layer(
              {base.IMAGE_KEY: image_tensor}
          )
          return model(transformed_features)
  
      return serve_tf_examples_fn
  
file_path: tfx/tfx/examples/mnist/mnist_utils_native_keras_lite.py
imports: import os
  import tensorflow
  import tensorflow_transform
  from tfx import v1
  from tfx.components.trainer.rewriting import converters
  from tfx.components.trainer.rewriting import rewriter
  from tfx.components.trainer.rewriting import rewriter_factory
  from tfx.examples.mnist import mnist_utils_native_keras_base
label_desc: Use image node_id 1 for calling a global function with example usage: _get_serve_tf_examples_fn(model, tf_transform_output) and returns: serve_tf_examples_fn, model