code: def tir_func(a: T.handle, b: T.handle) -> None:
      A = T.match_buffer(a, [2, 2])
      B = T.match_buffer(a, [2, 2])
      A[0, 1] = B[1, 1]
  
file_path: tvm/tests/python/tir-transform/test_tir_transform_storage_flatten.py
imports: import tvm
  import tvm.testing
  from tvm import te
  from tvm.driver.build_module import schedule_to_module
  from tvm.script import tir
  from tvm.relay import GlobalVar
label_desc: Use image node_id 6 for calling a global function with example usage: tir_func(a, b) without return types