code: def test_identity_single_removal_on_binary_elementwise():
      """Check that identity on the second input of the binary elementwise
      operation is removed while the other input has no identity."""
  
      def get_graph(get_expected=False):
          x = relay.var("x", shape=(1, 4, 1, 4), dtype="int8")
          y = relay.var("y", shape=(1, 2, 2, 4), dtype="int8")
          y = relay.reshape(y, newshape=(1, 4, 1, 4))
          if not get_expected:
              y = infra.make_ethosu_identity(y)
          z = infra.make_ethosu_binary_elementwise(
              x, y, 4, 4, "ADD", "int8"
          )
          return relay.Function(relay.analysis.free_vars(z), z)
  
      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 14 for calling a global function with example usage: test_identity_single_removal_on_binary_elementwise() and returns: relay