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

Fix a bug that multiple (conv, batch_norm) ops could not be optimized. #2187

Merged
merged 3 commits into from
Jun 15, 2023

Conversation

fatcat-z
Copy link
Collaborator

In such case, batch_norm op will be treated as consumers which will prevent from optimizing related(conv, batch_norm) ops. So update the logic to fix this.
Fix #2186

@fatcat-z fatcat-z requested a review from xiaowuhu June 14, 2023 10:42
@fatcat-z fatcat-z enabled auto-merge (squash) June 14, 2023 11:43
fatcat-z added 2 commits June 14, 2023 20:21
Signed-off-by: Jay Zhang <jiz@microsoft.com>
Signed-off-by: Jay Zhang <jiz@microsoft.com>
@fatcat-z fatcat-z merged commit b27aa05 into onnx:main Jun 15, 2023
@fatcat-z fatcat-z deleted the fuse_bn branch June 15, 2023 04:04
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.

BatchNorm cannot be fused in some cases
2 participants