def METHOD_NAME(
    data_shape,
    out_channels,
    kernel_size,
    strides,
    padding,
    dilation,
    groups,
    data_layout,
    kernel_layout,
    dtype,
):
    data = relay.var("data", relay.TensorType(data_shape, dtype))
    weight = relay.var("weight")
    net = relay.nn.conv2d(
        data=data,
        weight=weight,  # conv kernel
        strides=strides,
        padding=padding,
        dilation=dilation,
        groups=groups,
        channels=out_channels,
        kernel_size=kernel_size,
        data_layout=data_layout,
        kernel_layout=kernel_layout,
    )
    net = relay.add(net, net)
    net = relay.nn.relu(net)
    inputs = relay.analysis.free_vars(net)
    return relay.Function(inputs, net)