def METHOD_NAME(outs, ins, input_layout: str):
    """Schedule for input layout ncw-32c64w and output layout ncw"""
    func = te.create_prim_func([ins, outs])
    s = tir.Schedule(func)
    sum_block = s.get_block("sum")
    # Input is multiple of fixed chunk but output is NxCx1
    # Hence transform_layout is only applied on input
    input_transformed_layout = get_layout_transform_fn(input_layout)
    s.transform_layout(sum_block, buffer=("read", 0), index_map=input_transformed_layout)
    return s