experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Disable function inlining in RunGrappler() as it may cause restore graphs to be removed.

Browse Source 
PiperOrigin-RevId: 353744153
Change-Id: I5a4fa48954ce24ed16fa828287fbf204b07b3dba
This commit is contained in:
Kuangyuan Chen 2021-01-25 15:31:05 -08:00 committed by TensorFlower Gardener
parent 6f3bc3a09e
commit 7a9dada711
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/compiler/mlir/tensorflow/translate/upgrade_graph.cc
View File

@ -121,11 +121,16 @@ Status RunGrappler(MetaGraphDef* meta_graph_def) {
config_proto.mutable_experimental()->set_use_tfrt(true); config_proto.mutable_experimental()->set_use_tfrt(true);
config_proto.mutable_graph_options() config_proto.mutable_graph_options()
->mutable_optimizer_options() ->mutable_optimizer_options()
->set_do_function_inlining(true); ->set_do_function_inlining(false);
// Do not skip grappler optimization even for small graphs. // Do not skip grappler optimization even for small graphs.
config_proto.mutable_graph_options() config_proto.mutable_graph_options()
->mutable_rewrite_options() ->mutable_rewrite_options()
->set_min_graph_nodes(-1); ->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; grappler::ItemConfig item_config;
item_config.ignore_user_placement = false; item_config.ignore_user_placement = false;