Skip to content

Fuse conv_2d_nchw_fchw + pooling_nchw_sum into a single conv_2d_nchw_fchw#3127

Open
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_938088988
Open

Fuse conv_2d_nchw_fchw + pooling_nchw_sum into a single conv_2d_nchw_fchw#3127
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_938088988

Conversation

@copybara-service

Copy link
Copy Markdown
Contributor

Fuse conv_2d_nchw_fchw + pooling_nchw_sum into a single conv_2d_nchw_fchw

Gemini-generated doc describing the proof of equivalence: https://www.overleaf.com/read/rbtmrtpscyxx#1cd977

This change also includes a fuzz-test to add further confidence in the equivalence.

Fixes #3123

@j2kun

j2kun commented Jun 25, 2026

Copy link
Copy Markdown
Collaborator

@Zohaib58 would you be willing to review this PR?

@copybara-service copybara-service Bot force-pushed the test_938088988 branch 2 times, most recently from 874424a to e5e6468 Compare June 25, 2026 22:24
…fchw

Gemini-generated doc describing the proof of equivalence: https://www.overleaf.com/read/rbtmrtpscyxx#1cd977

This change also includes a fuzz-test to add further confidence in the equivalence.

Fixes #3123

PiperOrigin-RevId: 938088988
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.

Performance Gap: conv+pool is not fused into one linear operation

1 participant