code: def write_v2_saved_model(
      module: tf.Module,
      tf_function: tf.types.experimental.GenericFunction,
      name: str,
      saved_model_dir: str,
  ) -> function.ConcreteFunction:
      """Writes `tf_function` under attr `name` of `module` to `saved_model_dir`."""
      concrete_fn = trace_and_update_module(
          module, tf_function, name, strip_control_dependencies=False
      )
      tf.saved_model.save(module, saved_model_dir)
      return concrete_fn
  
imports: from typing import Any, Dict, Iterable, Mapping, Tuple, Union
  import tensorflow
  from tensorflow_transform import annotators
  from tensorflow_transform import common_types
  from tensorflow_transform import graph_tools
  from tensorflow_transform import tf2_utils
  from tensorflow_transform import tf_utils
  from tensorflow_transform.py_func import pyfunc_helper
  from tensorflow_transform.saved import constants
  from tensorflow_transform.saved import saved_model_loader
  from tensorflow_transform.saved import saved_transform_io
  from tensorflow.python.eager import function
  from tensorflow.python.eager import wrap_function
  from tensorflow.python.framework import composite_tensor
  from tensorflow.python.ops import lookup_ops
  from tensorflow.python.util import object_identity
label_desc: Use image node_id 7 for calling a global function with example usage: write_v2_saved_model(module, tf_function, name, saved_model_dir) and returns: concrete_fn