From 7a9dada711d6a264a1dacdca46f80a6c48dc41c4 Mon Sep 17 00:00:00 2001 From: Kuangyuan Chen Date: Mon, 25 Jan 2021 15:31:05 -0800 Subject: [PATCH] Disable function inlining in RunGrappler() as it may cause restore graphs to be removed. PiperOrigin-RevId: 353744153 Change-Id: I5a4fa48954ce24ed16fa828287fbf204b07b3dba --- .../compiler/mlir/tensorflow/translate/upgrade_graph.cc | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tensorflow/compiler/mlir/tensorflow/translate/upgrade_graph.cc b/tensorflow/compiler/mlir/tensorflow/translate/upgrade_graph.cc index 4844c132d4d..ca0d3f8f619 100644 --- a/tensorflow/compiler/mlir/tensorflow/translate/upgrade_graph.cc +++ b/tensorflow/compiler/mlir/tensorflow/translate/upgrade_graph.cc @@ -121,11 +121,16 @@ Status RunGrappler(MetaGraphDef* meta_graph_def) { config_proto.mutable_experimental()->set_use_tfrt(true); config_proto.mutable_graph_options() ->mutable_optimizer_options() - ->set_do_function_inlining(true); + ->set_do_function_inlining(false); // Do not skip grappler optimization even for small graphs. config_proto.mutable_graph_options() ->mutable_rewrite_options() ->set_min_graph_nodes(-1); + // Disable function inlining because it may cause restore graphs to be removed + // as we optimize all graphs together. + config_proto.mutable_graph_options() + ->mutable_rewrite_options() + ->set_function_optimization(RewriterConfig::OFF); grappler::ItemConfig item_config; item_config.ignore_user_placement = false;