code: def te_argmax_idx_val():
      def f_combine(x, y):
          lhs = tvm.tir.Select((x[1] >= y[1]), x[0], y[0])
          rhs = tvm.tir.Select((x[1] >= y[1]), x[1], y[1])
          return lhs, rhs
  
      def f_identity(dtype0: tvm.DataType, dtype1: tvm.DataType):
          return tvm.tir.const(-1, dtype0), tvm.te.min_value(dtype1)
  
      argmax = te.comm_reducer(f_combine, f_identity, name="argmax")
  
      m = te.var("m")
      n = te.var("n")
      idx = te.placeholder((m, n), name="idx", dtype="int32")
      val = te.placeholder((m, n), name="val", dtype="float32")
      k = te.reduce_axis((0, n), "k")
      max_idx, max_val = te.compute(
          (m,),
          lambda i: argmax((idx[i, k], val[i, k]), axis=k),
          name="argmax",
      )
      return [idx, val, max_idx, max_val]
  
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 33 for calling a global function with example usage: te_argmax_idx_val() and returns: list, lhs,  rhs, tvm,  tvm