def METHOD_NAME(cfg, input, filter, strides, padding, dilation, out_dtype):
    prim_func = compute_tir_conv2d_nchw_oihw(input.shape, filter.shape, input.dtype)
    output = te.extern_primfunc([input, filter], prim_func, name="tir")
    return output