Add legacy user of //third_party/tensorflow/compiler/jit.

PiperOrigin-RevId: 733016653

Author: tensorflower-gardener

official/projects/panoptic/tasks/panoptic_maskrcnn.py

@@ -446,6 +446,7 @@ def aggregate_logs(self, state=None, step_outputs=None):
 
   def _reduce_semantic_metrics(self, logs: Dict[str, Any]):
     """Updates the per class and mean semantic metrics in the logs."""
+    assert self.segmentation_perclass_iou_metric is not None
     ious = self.segmentation_perclass_iou_metric.result()
     if self.task_config.segmentation_evaluation.report_per_class_iou:
       for i, value in enumerate(ious.numpy()):
@@ -454,6 +455,7 @@ def _reduce_semantic_metrics(self, logs: Dict[str, Any]):
 
   def _reduce_panoptic_metrics(self, logs: Dict[str, Any]):
     """Updates the per class and mean panoptic metrics in the logs."""
+    assert self.panoptic_quality_metric is not None
     result = self.panoptic_quality_metric.result()
     valid_thing_classes = result['valid_thing_classes']
     valid_stuff_classes = result['valid_stuff_classes']