message: Allows xprof to tag the timeline for the test with the test name.
diff: @@ -44,6 +44,7 @@ from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import tf_logging as logging
  +from tensorflow.python.profiler import trace
  from tensorflow.python.saved_model import builder
  from tensorflow.python.saved_model import load
  from tensorflow.python.saved_model import loader
  @@ -821,6 +822,7 @@ class TfTrtIntegrationTestBase(test_util.TensorFlowTestCase):
  return self._MakeSavedModelV1(run_params)
  def RunTest(self, run_params):
  + with trace.Trace(run_params.test_name):
  should_run, reason_for_skipping = self.ShouldRunTest(run_params)
  if not should_run:
  return self.skipTest(reason_for_skipping)
  @@ -876,8 +878,8 @@ class TfTrtIntegrationTestBase(test_util.TensorFlowTestCase):
  else:
  infer_saved_model_dir = saved_model_dir
  - # Run the inference graph, either using the converted graph or the original
  - # graph with convert_online == True.
  + # Run the inference graph, either using the converted graph or the
  + # original graph with convert_online == True.
  logging.info("Running final inference graph\n")
  result = self._RunGraph(run_params, infer_saved_model_dir, inputs_data,
  GraphState.INFERENCE)
  