code: def test_split(dtype, shape, splits, axis):
      """Compare Split output with TVM."""
      np.random.seed(0)
  
      outputs = []
      inputs = {
          "a": tvm.nd.array(
              np.random.randint(
                  np.iinfo(dtype).min,
                  np.iinfo(dtype).max + 1,
                  size=shape,
                  dtype=dtype,
              )
          )
      }
      for npu in [False, True]:
          model = _get_model(shape, dtype, splits, axis)
          mod = tei.make_module(model, {})
          output_count = (
              splits if isinstance(splits, int) else len(splits) + 1
          )
          outputs.append(
              tei.build_and_run(
                  mod,
                  inputs,
                  output_count,
                  {},
                  npu=npu,
                  additional_config_args={
                      "inline_non_compute_intensive_partitions": False
                  },
              )
          )
  
          tei.verify(outputs, dtype, 0)
  
file_path: tvm/tests/python/contrib/test_ethosn/test_split.py
imports: import numpy
  import pytest
  import tvm
  from tvm import relay
  from tvm.testing import requires_ethosn
  from .None import infrastructure
label_desc: Use image node_id 2 for calling a global function with example usage: test_split(dtype, shape, splits, axis) without return types