def METHOD_NAME(self):
    # Set the IGNORE_PERF_VAR variable back to its original value.
    if self._original_ignore_perf_value is not None:
        os.environ[self.IGNORE_PERF_VAR] = self._original_ignore_perf_value
    else:
        del os.environ[self.IGNORE_PERF_VAR]
    tf.compat.v1.mixed_precision.disable_mixed_precision_graph_rewrite()
    super().METHOD_NAME()