def METHOD_NAME(outs, ins, output_layout: str, input_layout: str):
    """avg_pool2d schedule"""
    if output_layout == "nhwc-8h2w32c2w-2d" or output_layout == "nchw-8h2w32c2w-2d":
        return schedule_8h2w32c2w(outs, ins, output_layout, input_layout)
    if output_layout == "n11c-1024c-2d" or output_layout == "nc11-1024c-2d":
        return schedule_1024c(outs, ins, output_layout, input_layout)
    raise RuntimeError(f"Unexpected layout '{output_layout}'")