code: def get_conv2d(data_shape, weight_shape, **kwargs):
      data = relay.var("data", shape=data_shape, dtype="float32")
      weight = relay.var("weight", shape=weight_shape, dtype="float32")
      conv2d = relay.nn.conv2d(
          data,
          weight,
          **kwargs,
      )
      return relay.Function([data, weight], conv2d)
  
file_path: tvm/tests/python/meta_schedule/test_meta_schedule_relay_tir_compute.py
imports: import numpy
  import tvm
  import tvm.testing
  import tvm.topi.testing
  from tvm import autotvm
  from tvm import meta_schedule
  from tvm import relay, te
  from tvm.relay.testing.temp_op_attr import TempOpAttr
  from tvm.script import tir
label_desc: Use image node_id 6 for calling a global function with example usage: get_conv2d(data_shape, weight_shape) and returns: relay