commit 09f617fe9f5e5c3844162b2de576ff420d8c80c6
author Renjie Liu <renjieliu@google.com> Thu Mar 05 10:09:17 2020 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> Thu Mar 05 10:12:57 2020 -0800
tree eff0477b06b8f457757fa0e206227742d00ab79a
parent 35274af2e04ed4855fe2845a6867d600b64071d8

Turn on saved_model related passes if we're importing from saved_model.

PiperOrigin-RevId: 299130053
Change-Id: I0f640db8c28ddec22bdfa9ed1187bf7edc92b0cd

tensorflow/compiler/mlir/lite/tf_tfl_translate.cc

diff --git a/tensorflow/compiler/mlir/lite/tf_tfl_translate.cc b/tensorflow/compiler/mlir/lite/tf_tfl_translate.cc
index 7f8ce4c..21712e7 100644
--- a/tensorflow/compiler/mlir/lite/tf_tfl_translate.cc
+++ b/tensorflow/compiler/mlir/lite/tf_tfl_translate.cc
@@ -195,6 +195,8 @@
   pass_config.emit_builtin_tflite_ops = emit_builtin_tflite_ops;
   pass_config.lower_tensor_list_ops = lower_tensor_list_ops;
   pass_config.inline_functions = inline_functions;
+  if (import_saved_model || import_saved_model_v1)
+    pass_config.saved_model_import = true;
 
   tensorflow::AddTFToTFLConversionPasses(pass_config, &pm);
   pm.addPass(mlir::TFL::CreateRuntimeTypeVerifyPass());