code: def _get_model(shape, dtype, splits, axis):
      a = relay.var("a", shape=shape, dtype=dtype)
      split = relay.op.split(a, indices_or_sections=splits, axis=axis)
      return split.astuple()
  
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 1 for calling a global function with example usage: _get_model(shape, dtype, splits, axis) and returns: split