code: def test_if():
      x = relay.var("x", shape=(10, 10))
      y = relay.var("y", shape=(10, 10))
      equal = relay.op.equal(x, y)
      equal = relay.op.nn.batch_flatten(equal)
      f = relay.Function(
          [x, y], relay.If(relay.op.min(equal, axis=[0, 1]), x, y)
      )
      x_data = np.random.rand(10, 10).astype("float32")
      y_data = np.random.rand(10, 10).astype("float32")
  
      # same
      res = get_serialized_output(f, x_data, x_data)
      tvm.testing.assert_allclose(res.numpy(), x_data)
  
      # diff
      res = get_serialized_output(f, x_data, y_data)
      tvm.testing.assert_allclose(res.numpy(), y_data)
  
imports: import pytest
  import numpy
  import tvm
  from tvm.runtime import vm
  from tvm.relay import vm
  from tvm import relay
  from tvm.relay.scope_builder import ScopeBuilder
  from tvm.relay import transform
  from tvm.relay.prelude import Prelude
  from tvm.contrib import utils
  from tvm.relay import testing
label_desc: Use image node_id 7 for calling a global function with example usage: test_if() without return types