code: def _create_tf2_saved_model(self):
      model_dir = os.path.join(self.get_temp_dir(), "saved_model")
      model = MyModel()
  
      @tf.function
      def serving_default(input_):
          return {"output": model(input_)}
  
      signature_function = serving_default.get_concrete_function(
          tf.TensorSpec(
              shape=[
                  3,
              ],
              dtype=tf.float32,
          )
      )
      tf.saved_model.save(
          model,
          model_dir,
          signatures={
              tf.saved_model.DEFAULT_SERVING_SIGNATURE_DEF_KEY: signature_function
          },
      )
  
      return model_dir
  
file_path: hub/tensorflow_hub/saved_model_module_test.py
imports: import os
  import tensorflow
  import tensorflow_hub
label_desc: Use image node_id 1 for calling the SavedModelTest obj's underlying member method code with example usage: obj._create_tf2_saved_model() and returns: model_dir, dict