Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Insert Identity nodes on given tensor and top-level graph inputs #166

Merged
merged 6 commits into from
Jan 13, 2025

Conversation

maltanar
Copy link
Collaborator

Introduces transformations that insert Identity nodes in the ONNX graph at desired locations, which can either be a specified tensor (InsertIdentity) or all top-level input/output tensors (InsertIdentityOnAllTopLevelIO), as well as corresponding unit tests. This can be useful before calling transformations that do not gracefully handle edge cases where transformed tensors are top-level inputs or outputs.

Original graph:
image

After InsertIdentityOnAllTopLevelIO:
image

@maltanar maltanar merged commit 530f3e2 into main Jan 13, 2025
5 checks passed
@maltanar maltanar deleted the feature/insert_identity branch January 13, 2025 11:28
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant