code: def test_reshape_last():
      """Check that an identity as a leaf of the graph is not removed."""
  
      def get_graph():
          x = relay.var("x", shape=(1, 2, 2, 4), dtype="int8")
          x = infra.make_ethosu_conv2d(
              x, 4, 4, (1, 1), (0, 0), (1, 1), (1, 1)
          )
          x = relay.reshape(x, newshape=(1, 4, 4, 1))
          x = infra.make_ethosu_identity(x)
          return relay.Function(relay.analysis.free_vars(x), x)
  
      actual = _optimize(get_graph())
      expected = _optimize(get_graph(), optimize=False)
      _assert_structural_equal(actual, expected)
  
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 6 for calling a global function with example usage: test_reshape_last() and returns: relay