commit | d583b619207aedb9c1642a2a58e1b626bcbebb1d | [log] [tgz] |
---|---|---|
author | A. Unique TensorFlower <gardener@tensorflow.org> | Thu Jan 28 16:58:05 2021 -0800 |
committer | TensorFlower Gardener <gardener@tensorflow.org> | Thu Jan 28 17:11:30 2021 -0800 |
tree | d17aaeae80fc535cfb3a642923fad1036b79b7d8 | |
parent | 1aa07125d5d1ca135aa03779dceee064f89a2828 [diff] |
Disable shape inference for TFRT importing SavedModel. PiperOrigin-RevId: 354426742 Change-Id: I3203df9423ffd4bf69f9cc4ccff1a63927d7599f
diff --git a/tensorflow/compiler/mlir/tensorflow/translate/import_model.cc b/tensorflow/compiler/mlir/tensorflow/translate/import_model.cc index c0110ca..54c095f 100644 --- a/tensorflow/compiler/mlir/tensorflow/translate/import_model.cc +++ b/tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
@@ -3555,7 +3555,6 @@ GraphImportConfig specs; specs.prune_unused_nodes = true; specs.inputs = ParseInputArrays(inputs); - specs.enable_shape_inference = false; for (auto& output : outputs) specs.outputs.push_back(output.second.name()); specs.control_outputs = control_outputs;