code: def test_argmax():
      data = relay.var("data", shape=(1, 64, 56, 56), dtype="uint8")
      mod = tvm.IRModule.from_expr(relay.argmax(data, axis=1))
  
      target = tvm.target.Target("llvm")
  
      opt_mod, _ = relay.optimize(mod, params={}, target=target)
  
      prim_func = relay.backend.te_compiler.lower_to_primfunc(
          opt_mod["main"].body.op, target
      )
  
      tvm.ir.assert_structural_equal(prim_func, argmax_expected)
  
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 45 for calling a global function with example usage: test_argmax() without return types