code: def test_add_one_2d_constant():
      class AddOne2DConstant(tf.Module):
          """2D array as input with 2D constant as well; 2D constant stored in params after convert"""
  
          def get_input(self):
              return np.ones((2, 2), dtype="float32")
  
          @tf.function(
              input_signature=[
                  tf.TensorSpec(shape=(2, 2), dtype=tf.float32)
              ]
          )
          def func(self, x):
              return x + np.ones((2, 2), dtype="float32")
  
      run_all(AddOne2DConstant)
  
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 8 for calling a global function with example usage: test_add_one_2d_constant() and returns: np, unknown