code: def resize_cache_index(
      A: T.Buffer((1, 3, 40, 40), "float32"),
      B: T.Buffer((1, 3, 80, 80), "float32"),
  ) -> None:
      index_var_0 = T.alloc_buffer([80, 80], dtype="int32", strides=[1])
      index_var_1 = T.alloc_buffer([80], dtype="int32", strides=[1])
      for ax0, ax1 in T.grid(80, 80):
          with T.block("index_0"):
              v0, v1 = T.axis.remap("SS", [ax0, ax1])
              T.reads()
              T.writes(index_var_0[v0, v1])
              index_var_0[v0, v1] = v0 // 4 + v1 // 4
      for ax0 in T.serial(80):
          with T.block("index_1"):
              v0 = T.axis.spatial(80, ax0)
              T.reads()
              T.writes(index_var_1[v0])
              index_var_1[v0] = v0 // 2
      for i0, i1, i2, i3 in T.grid(1, 3, 80, 80):
          with T.block("A"):
              n, c, vi, vj = T.axis.remap("SSSS", [i0, i1, i2, i3])
              T.reads(A[n, c, vi // 4 + vj // 4, vj // 2])
              T.writes(B[n, c, vi, vj])
              B[n, c, vi, vj] = A[
                  n, c, index_var_0[vi, vj], index_var_1[vj]
              ]
  
file_path: tvm/tests/python/tir-schedule/test_tir_schedule_cache_index.py
imports: import sys
  import pytest
  import tvm
  import tvm.testing
  from tvm import tir
  from tvm.script import tir
  from tvm.tir.schedule.testing import verify_trace_roundtrip
label_desc: Use image node_id 2 for calling a global function with example usage: resize_cache_index(A, B) without return types