code: def _model_graph(TestClass):
      model = TestClass()
      with tempfile.TemporaryDirectory() as model_path:
          tf.saved_model.save(model, model_path)
          imported = tf.saved_model.load(model_path)
  
      f = imported.signatures["serving_default"]
      gdef = f.graph.as_graph_def(add_shapes=True)
  
      input_ = model.get_input()
      output = run_tf_code(f, input_)
      return gdef, input_, output
  
imports: import tempfile
  import tensorflow
  import numpy
  import pytest
  from common import compare_tf_tvm
  from common import run_tf_code
label_desc: Use image node_id 2 for calling a global function with example usage: _model_graph(TestClass) and returns: gdef,  input_,  output