class_docstr: Argmax Slice Op Tests
docstr: Top level testing function for argmax
docstr_tok: ["Top","level","testing","function","for","argmax"]
code: def test_argmax_slice(
      self,
      input_shape,
      dtype,
      input_layout,
      output_layout,
      in_axis,
      transformed_input_np,
      transformed_expected_output_np,
      in_axis_sep,
      out_axis_sep,
      hexagon_session,
      working_scope,
  ):
      """Top level testing function for argmax"""
      argmax_input = te.placeholder(input_shape, name="A", dtype=dtype)
      output = sl.argmax.argmax_compute(argmax_input, in_axis)
      argmax_func = te.create_prim_func([argmax_input, output])
      tir_s = sl.argmax_schedule(
          argmax_func, input_layout, output_layout
      )
      input_data = allocate_hexagon_array(
          hexagon_session.device,
          data=transformed_input_np,
          axis_separators=in_axis_sep,
          mem_scope=working_scope,
      )
      output_data = allocate_hexagon_array(
          hexagon_session.device,
          tensor_shape=transformed_expected_output_np.shape,
          dtype=transformed_expected_output_np.dtype,
          axis_separators=out_axis_sep,
          mem_scope=working_scope,
      )
      with tvm.transform.PassContext(opt_level=3):
          tir_irm = tvm.lower(
              tir_s.mod, [argmax_input, output], name="argmax"
          )
          runtime_module = tvm.build(
              tir_irm,
              [argmax_input, output],
              target=get_hexagon_target("v69"),
              name="argmax",
          )
      mod = hexagon_session.load_module(runtime_module)
  
      mod(input_data, output_data)
      output_np = output_data.numpy()
      tvm.testing.assert_allclose(
          output_np,
          transformed_expected_output_np,
          1e-3,
          1e-3,
      )
  
file_path: tvm/tests/python/contrib/test_hexagon/topi/slice_op/test_argmax_slice.py
imports: import numpy
  import tvm
  import tvm.testing
  from tvm import te
  import tvm.topi.hexagon.slice_ops
  import tvm.contrib.hexagon
  from tvm.contrib.hexagon import allocate_hexagon_array
  from ...infrastructure import transform_numpy, get_hexagon_target
label_desc: Use image node_id 5 for calling the TestArgMaxSlice obj's underlying member method code with example usage: obj.test_argmax_slice(input_shape, dtype, input_layout, output_layout, in_axis, transformed_input_np, transformed_expected_output_np, in_axis_sep, out_axis_sep, hexagon_session, working_scope) without return types