class_docstr: Wraps a function and its parameters for saving to a SavedModel.
  
  Implements the interface described at
  https://www.tensorflow.org/hub/reusable_saved_models.
  
  This class contains all the code needed to convert a Flax model to a
  TensorFlow saved model.
code: def __call__(self, inputs):
      return self._tf_graph(inputs)
  
imports: import functools
  from typing import Callable, Dict, List
  import absl
  from flax import linen
  from flax.metrics import tensorboard
  import jax
  from jax import numpy
  from jax.experimental import jax2tf
  import numpy
  import optax
  import tensorflow
  import tensorflow_transform
  from tfx import v1
  from tfx.examples.penguin import penguin_utils_base
label_desc: Use image node_id 2 for calling the _SavedModelWrapper obj's underlying member method code with example usage: obj.__call__(inputs) and returns: self