code: def get_conv2d():
      x = relay.var("x", shape=(1, 56, 56, 64))
      weight1 = relay.var("weight1", shape=(3, 3, 64, 32))
      y = relay.nn.conv2d(
          x,
          weight1,
          channels=32,
          kernel_size=(3, 3),
          padding=(1, 1),
          data_layout="NHWC",
          kernel_layout="HWIO",
      )
      return tvm.IRModule.from_expr(y)
  
imports: import pytest
  import tvm
  from tvm import relay
label_desc: Use image node_id 2 for calling a global function with example usage: get_conv2d() and returns: tvm