code: def test_identity_removal_with_multiple_transform_ops():
      """Check that only an identity directly parent to a compute
      operation is removed."""
  
      def get_graph(get_expected=False):
          x = relay.var("x", shape=(1, 2, 2, 4), dtype="int8")
          x = relay.strided_slice(
              x, begin=[0, 0, 0, 0], end=[1, 2, 2, 2]
          )
          if not get_expected:
              x = infra.make_ethosu_identity(x)
          x = relay.reshape(x, newshape=(1, 1, 1, 8))
          if not get_expected:
              x = infra.make_ethosu_identity(x)
          x = infra.make_ethosu_unary_elementwise(x, 8, "ABS")
          return relay.Function(relay.analysis.free_vars(x), x)
  
      actual = _optimize(get_graph())
      expected = _optimize(get_graph(get_expected=True), 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 12 for calling a global function with example usage: test_identity_removal_with_multiple_transform_ops() and returns: relay