code: def test_multi_output_identity_has_same_output():
      """Check that the output remains the same with an identity with
      multiple outputs."""
      ifm_shape = (1, 1, 64, 16)
  
      @tf.function
      def model(x):
          x = tf.reshape(x, (1, 8, 8, 16))
          outputs = []
          for _ in range(4):
              outputs.append(tf.nn.max_pool2d(x, 1, 1, "VALID"))
          outputs.append(tf.reshape(x, (1, 8, 8, 16)))
          y = tf.concat(outputs, axis=0)
          return y
  
      infra.compare_tvm_with_tflite(
          model, [ifm_shape], "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 18 for calling a global function with example usage: test_multi_output_identity_has_same_output() and returns: y