def METHOD_NAME():
    mod = tvm.IRModule({})
    # NPU function
    x = relay.var("x", shape=(1, 4, 4, 16), dtype="int8")
    max_pool = relay.nn.max_pool2d(x)
    composite_func = relay.Function([x], max_pool)
    composite_func = composite_func.with_attr("Composite", "ethos-u.pooling")
    inp = relay.var("input", shape=(1, 4, 4, 16), dtype="int8")
    compiler_func = relay.Function([inp], composite_func)
    compiler_func = compiler_func.with_attr("used_memory", [256 + 256])
    npu_compiler_func = compiler_func.with_attr("Compiler", "ethos-u")
    g1 = relay.GlobalVar("g1")
    mod[g1] = npu_compiler_func
    # Main
    inp = relay.var("main_input", shape=(1, 4, 4, 16), dtype="int8")
    call = relay.Call(g1, [inp])
    main_func = relay.Function([inp], call)
    main_func = main_func.with_attr("io_used_memory", 256 + 256)
    mod["main"] = main_func
    return mod