commit 5d7f57bd0b7d29e0fb287052f37e460d67f3725c
author Andy Ly <lyandy@google.com> Thu Dec 10 12:02:48 2020 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> Thu Dec 10 12:19:59 2020 -0800
tree 927cc2bbe47d2922fa4180ffebb8c4e038247c8c
parent 2bec62824f1a7c5b84b2ca7dead5c220bdd71f89

Disable shape inference on import in ConvertFunctionToMlir.

Shape inference on import is deprecated in favor of shape inference pass.

PiperOrigin-RevId: 346835235
Change-Id: I38e67f9b87442d00680ad5d76e5d8c74e9ca7f69

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 4d33308..53c7355 100644
--- a/tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
+++ b/tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
@@ -3733,6 +3733,7 @@
     mlir::MLIRContext* context) {
   tensorflow::GraphDebugInfo dummy_debug_info;
   tensorflow::GraphImportConfig specs;
+  specs.enable_shape_inference = false;
   specs.graph_as_function = true;
   for (const auto* control_ret_node : fbody->control_ret_nodes)
     specs.control_outputs.push_back(control_ret_node->name());