code: def tir_argmax_idx_val(
      var_idx: T.handle,
      var_val: T.handle,
      var_argmax_v0: T.handle,
      var_argmax_v1: T.handle,
  ) -> None:
      T.func_attr({"global_symbol": "main", "tir.noalias": True})
      m = T.int32()
      n = T.int32()
      idx = T.match_buffer(var_idx, [m, n], dtype="int32")
      val = T.match_buffer(var_val, [m, n], dtype="float32")
      argmax_v0 = T.match_buffer(var_argmax_v0, [m], dtype="int32")
      argmax_v1 = T.match_buffer(var_argmax_v1, [m], dtype="float32")
      for i0, i1 in T.grid(m, n):
          with T.block("argmax"):
              i, k = T.axis.remap("SR", [i0, i1])
              T.reads(val[i, k], idx[i, k])
              T.writes(argmax_v0[i], argmax_v1[i])
              with T.init():
                  argmax_v0[i] = T.int32(-1)
                  argmax_v1[i] = T.min_value("float32")
              v_argmax_v0: T.int32 = T.Select(
                  argmax_v1[i] >= val[i, k], argmax_v0[i], idx[i, k]
              )
              v_argmax_v1: T.float32 = T.Select(
                  argmax_v1[i] >= val[i, k], argmax_v1[i], val[i, k]
              )
              argmax_v0[i] = v_argmax_v0
              argmax_v1[i] = v_argmax_v1
  
imports: import numpy
  import tvm
  import tvm.testing
  from tvm import te, tir, topi, relay
  from tvm.script import tir
  import pytest
label_desc: Use image node_id 34 for calling a global function with example usage: tir_argmax_idx_val(var_idx, var_val, var_argmax_v0, var_argmax_v1) without return types