Performs TensorFlow shape inference after inlining. * Allows more shapes to be discovered that may have been initially computed as dynamic due to TensorFlow's function rules. * Requires bumping tensorflow to 15ef640d4f7e5cdb78aa3b67a8b9d98de30e9e94 (populates tf.versions from SavedModel) to fully function but will just no-op without that patch. PiperOrigin-RevId: 289170622
diff --git a/bindings/python/pyiree/compiler/conditional_tensorflow.py b/bindings/python/pyiree/compiler/conditional_tensorflow.py index 5e6034c..4035781 100644 --- a/bindings/python/pyiree/compiler/conditional_tensorflow.py +++ b/bindings/python/pyiree/compiler/conditional_tensorflow.py
@@ -41,7 +41,10 @@ # Clean up control flow "tf-functional-control-flow-to-cfg", "inline", + + # Some further cleanups now that control flow is in better shape. "tf-saved-model-delete-unused-funcs", + "tf-shape-inference", "canonicalize", # Legalize to XLA