code: def stir_global_avg_pool2d_u8_schedule(
      outs: te.Tensor, ins: te.Tensor, input_layout: str
  ):
      """Schedule"""
      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 NxCx1x1
      # 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
  
imports: from tvm import te
  from tvm import tir
  from ..utils import get_layout_transform_fn, get_fixed_point_value, saturate
label_desc: Use image node_id 2 for calling a global function with example usage: stir_global_avg_pool2d_u8_schedule(outs, ins, input_layout) and returns: s