code: def test_ethosu_softmax(
      accel_type,
      ifm_shape,
  ):
      np.random.seed(0)
  
      @tf.function
      def softmax(x):
          return tf.nn.softmax(x)
  
      infra.compare_tvm_with_tflite(
          softmax, [ifm_shape], accel_type, ranges=[(-1, 1)]
      )
  
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 13 for calling a global function with example usage: test_ethosu_softmax(accel_type, ifm_shape) and returns: tf