code: def avg_pool2d_schedule(
      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}'")
  
file_path: tvm/python/tvm/topi/hexagon/slice_ops/avg_pool2d.py
imports: from tvm import te
  from tvm import tir
  from ..utils import get_layout_transform_fn
  from ...utils import get_const_tuple
  from ...nn.utils import get_pad_tuple
  from ...nn.pad import pad
  from ..compute_poolarea import compute_PoolArea
label_desc: Use image node_id 5 for calling a global function with example usage: avg_pool2d_schedule(outs, ins, output_layout, input_layout) and returns: schedule_8h2w32c2w, schedule_1024c