code: def compute(self, use_jit, compute_fn):
      random_seed.set_random_seed(1234)
      with self.session(graph=ops.Graph()) as sess:
          with jit.experimental_jit_scope(use_jit):
              r = compute_fn()
          sess.run(variables.global_variables_initializer())
          return (r, sess.run(r))
  
imports: from absl.testing import parameterized
  from tensorflow.python.compiler.xla import jit
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import function
  from tensorflow.python.framework import op_def_registry
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import random_seed
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import gradients
  from tensorflow.python.ops import init_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import random_ops
  from tensorflow.python.ops import variable_scope
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the JITTest obj's underlying member method code with example usage: obj.compute(use_jit, compute_fn) and returns: r,  sess