commit 586a791d502921a31fbfff30f3147242aa5b3579
author Roman Dzhabarov <rdzhabarov@google.com> Fri Jul 30 15:10:05 2021 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Fri Jul 30 15:15:04 2021 -0700
tree 4e62a9adaf622a906c101549d78a6e6a03818884
parent 35639cd25a878ecd7b97c09b48fbcd7119540db2

[mlir/import_model] Save only flib reachable from the graph def so that it can be reported in case of a crash.

PiperOrigin-RevId: 387891613
Change-Id: I4aee462527c72aa382fb48829b96fac2221c3158

tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

diff --git a/tensorflow/compiler/mlir/tensorflow/translate/import_model.cc b/tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
index 5af156e..f397975 100644
--- a/tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
+++ b/tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
@@ -2339,9 +2339,10 @@
   const auto* graph_crash_handle = crash_analysis::ReportProtoDataOnCrash(
       absl::StrCat(current_file_prefix, "_mlir_import_graph.pbtxt"),
       *graph_def);
+  auto reachable_flib = flib_def.ReachableDefinitions(*graph_def);
   const auto* flib_crash_handle = crash_analysis::ReportProtoDataOnCrash(
       absl::StrCat(current_file_prefix, "_mlir_import_flib.pbtxt"),
-      flib_def.ToProto());
+      reachable_flib.ToProto());
 
   auto scope_exit = llvm::make_scope_exit([&]() {
     crash_analysis::RemoveReportData(graph_crash_handle);