commit d1e617ded2a7b172c89529b65a646eb7c9a86d37
author Kuangyuan Chen <chky@google.com> Fri Aug 07 16:14:53 2020 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Fri Aug 07 16:20:00 2020 -0700
tree 28df07bc55db30091ef0621c55941066df795910
parent a8dad139b6a7ecce56cfad1a2838e31d4870de1e

Pass `upgrade_legacy` cmd line flag to signaturedef importer in tf-mlir-translate.

PiperOrigin-RevId: 325528613
Change-Id: I43585f35c9a7038bd0c094d3ffb68df8103a3a9f

tensorflow/compiler/mlir/tf_mlir_translate_main.cc

diff --git a/tensorflow/compiler/mlir/tf_mlir_translate_main.cc b/tensorflow/compiler/mlir/tf_mlir_translate_main.cc
index 8cfdfd0..caac8ea 100644
--- a/tensorflow/compiler/mlir/tf_mlir_translate_main.cc
+++ b/tensorflow/compiler/mlir/tf_mlir_translate_main.cc
@@ -121,7 +121,7 @@
     mlir::MLIRContext context;
 
     auto module_or = tensorflow::SavedModelSignatureDefsToMlirImport(
-        input_filename, tags, exported_names, &context);
+        input_filename, tags, exported_names, &context, upgrade_legacy);
     if (!module_or.status().ok()) return 1;
 
     module_or.ConsumeValueOrDie()->print(output->os());