code: def relu_te_sched(Output, Input, layout):
      """
      Schedule assumes the layout function to be bijective
      """
      s = te.create_schedule(Output.op)
      s[Input].transform_layout(layout)
      out_axes = s[Output].transform_layout(layout)
      fused = s[Output].fuse(out_axes[6], out_axes[7])
      s[Output].vectorize(fused)
      return s
  
imports: from tvm import te, tir, topi
  from ..utils import get_layout_transform_fn
label_desc: Use image node_id 2 for calling a global function with example usage: relu_te_sched(Output, Input, layout) and returns: s