code: def schedule_1024c(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
  
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 4 for calling a global function with example usage: schedule_1024c(outs, ins, output_layout, input_layout) and returns: s