code: def test_tflite_split(
      accel_type, ifm_shape, num_or_size_splits, axis
  ):
      np.random.seed(0)
  
      @tf.function
      def split_func(x):
          op = tf.split(x, num_or_size_splits, axis=axis)
          return op
  
      infra.compare_tvm_with_tflite(
          split_func, [ifm_shape], accel_type, enable_cascader=False
      )
  
imports: import pytest
  import numpy
  import tflite.Model
  import tvm
  import tensorflow
  from tvm import relay
  from tvm.relay.backend.contrib.ethosu import util
  from tvm.relay.op.contrib.ethosu import partition_for_ethosu
  from tvm.testing.aot import generate_ref_data
  from .None import infra
label_desc: Use image node_id 36 for calling a global function with example usage: test_tflite_split(accel_type, ifm_shape, num_or_size_splits, axis) and returns: op