Preserve disconnected Constant->Result subgraphs in TensorFlow models #32269
+451
−4
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.
Details:
After CVS-136813, TensorFlow Frontend stopped removing Constant->Result subgraphs that represent training artifacts (e.g., Adam optimizer parameters like beta_1, beta_2). However, the test
ModelWithConstResultSubgraphs
was failing because:ConstToResultRemover
was incorrectly removing all Constant->Result subgraphs, not just UnsupportedConstant->ResultFixed ConstToResultRemover transformation
src/frontends/tensorflow_common/src/helper_transforms/const_to_result_remover.cpp
Removed logic that deleted all Constant->Result subgraphs. Now only removes UnsupportedConstant->Result subgraphs as intended. This preserves legitimate training artifacts from TensorFlow models.
Updated reference model
src/frontends/tensorflow/tests/convert_tricky_models.cpp
Added disconnected Adam/beta_1 and Adam/beta_2 constants to the reference model to match the actual TensorFlow model structure.
Tickets: