if (func.getName().starts_with("__inference__traced_save_") ||
        func.getName().starts_with("__inference__traced_restore_") ||
        func.getName().starts_with("__inference_signature_wrapper_")) {
      func.erase();
    }
  }

  // Diagnose SavedFunction's with multiple input signatures.
  TF_RETURN_IF_ERROR(
      DiagnoseMultipleConcreteFunctions(object_graph, object_names));