code: def testBasicInt32(self):
      with ops.Graph().as_default() as g:
  
          def fn(x, a):
              return x + a
  
          xla_func = def_function.function(fn, jit_compile=True)
          inputs = array_ops.placeholder(dtypes.int32, [5])
          x = xla_func(inputs, 1)
          with session.Session(graph=g) as sess:
              y = sess.run(x, feed_dict={inputs: [1, 2, 2, 3, 3]})
              self.assertTrue(
                  x.graph.as_graph_def()
                  .library.function[0]
                  .attr["_XlaMustCompile"]
                  .b
              )
              self.assertAllClose([2, 3, 3, 4, 4], y)
  
file_path: tensorflow/tensorflow/python/compiler/xla/tests/jit_compile_test.py
imports: from tensorflow.python.client import session
  from tensorflow.python.eager import backprop
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import string_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 3 for calling the JitCompileTest obj's underlying member method code with example usage: obj.testBasicInt32() and returns: unknown