code: def test_same_output():
      """Check that the output remains the same when the identity
      optimizer pass removes some identities inserted during legalization.
      """
      ifm_shapes = [(1, 1, 25, 8), (1, 5, 5, 8)]
  
      @tf.function
      def model(x, y):
          x = tf.reshape(x, (1, 5, 5, 8))
          z = tf.add(x, y)
          z = tf.reshape(z, (1, 1, 25, 8))
          return z
  
      infra.compare_tvm_with_tflite(
          model, ifm_shapes, "ethos-u55-256", enable_cascader=False
      )
  
imports: import pytest
  import tensorflow
  import tvm
  from tvm import relay
  from tvm.relay.op.contrib.ethosu import partition_for_ethosu
  from tvm.relay.backend.contrib.ethosu.codegen import relay_to_tir
  from tvm.relay.backend.contrib.ethosu.codegen import IdentityOptimizer
  from .None import infra
label_desc: Use image node_id 17 for calling a global function with example usage: test_same_output() and returns: z