code: def simpleTest(self, arg0, arg1, global_jit_level):
      config = config_pb2.ConfigProto()
      config.graph_options.optimizer_options.global_jit_level = (
          global_jit_level
      )
  
      with session_lib.Session(config=config) as sess:
          a1 = array_ops.placeholder(dtypes.float32, [2, 2], name="a1")
          a2 = array_ops.placeholder(dtypes.float32, [2, 2], name="a2")
          # Two element-wise ops. We need at least two ops since single
          # element clusters are not passed to XLA in fusion_only mode.
          a3 = a1 * a2
          a4 = a3 + a1
          # A matmul to break XLA clustering.
          a5 = math_ops.matmul(a4, a1)
          # Two more element-wise ops.
          a6 = a5 - a4
          a7 = a6 + a2
  
          run_metadata = config_pb2.RunMetadata()
          output = test_utils.RunWithWarmup(
              sess,
              a7,
              {a1: arg0, a2: arg1},
              run_metadata=run_metadata,
              options=config_pb2.RunOptions(
                  trace_level=config_pb2.RunOptions.FULL_TRACE
              ),
          )
  
          labels = RunMetadataLabels(run_metadata)
  
          xla_compile_count = sum("XlaCompile(" in x for x in labels)
          xla_run_count = sum("XlaRun(" in x for x in labels)
          self.assertEqual(xla_compile_count, xla_run_count)
  
          return output, xla_run_count
  
imports: import os
  import numpy
  from tensorflow.compiler.tests import test_utils
  from tensorflow.core.protobuf import config_pb2
  from tensorflow.core.protobuf import rewriter_config_pb2
  from tensorflow.python.client import session
  from tensorflow.python.compiler.xla import jit
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import function
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import cond
  from tensorflow.python.ops import control_flow_ops
  from tensorflow.python.ops import gradients_impl
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.ops import while_loop
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the ElementWiseFusionTest obj's underlying member method code with example usage: obj.simpleTest(arg0, arg1, global_jit_level) and returns: output,  xla_run_count