def METHOD_NAME(outs, ins, output_layout: str, input_layout: str):
    """Schedule for input and output layout 8h2w32c2w"""
    func = te.create_prim_func([ins, outs])
    print(func)
    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)
    return s