Tweak rewrites in infer_shape
dataset
#1563
Merged
+1
−2
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This PR removes some inc_subtensor rewrites from infer_shape database. These don't simplify shape inference as the shape of an inc_subtensor is always the shape of the input, regardless of whether it is useless or not. They are also not very common to begin with?
Instead we add the useless_switch rewrite. We want a subset of rewrites that quickly simplify the graph and reason about the shapes introduced by
infer_shape
📚 Documentation preview 📚: https://pytensor--1563.org.readthedocs.build/en/1563/