def METHOD_NAME(outs, ins, output_layout: str, input_layout: str):
    """Schedule for output layout: 1024c, input layout: 8h2w32c2w"""
    func = te.create_prim_func([ins, outs])
    s = tir.Schedule(func)
    Sum = s.get_block("pool_sum")
    Avg = s.get_block("pool_avg")
    mem_scope = "global.vtcm"
    sum_read = s.cache_read(Sum, 0, mem_scope)
    avg_write = s.cache_write(Avg, 0, mem_scope)
    input_transform_fn = get_layout_transform_fn(input_layout)
    output_transform_fn = get_layout_transform_fn(output_layout)
    s.transform_layout(Sum, ("read", 0), input_transform_fn, pad_value=0.0)
    s.transform_layout(Avg, ("write", 0), output_transform_fn, pad_value=0.0)
    # Schedule 'Avg'
    if output_layout == "n11c-1024c-2d":
        n, h, w, c = s.get_loops(Avg)
    else:
        n, c, h, w = s.get_loops(Avg)
    _, ci = s.split(c, [None, 1024])
    cio, cii = s.split(ci, [None, 64])
    s.vectorize(cii)
    # Schedule 'Sum'
    Sum_axis = s.get_loops(Sum)
    s.reorder(Sum_axis[-2], Sum_axis[-1], Sum_axis[-3])
    return s