Commit 215fa1b
Fix broken ConvBNReLu from new Convert1DConvTo2D pass (#19558)
Summary:
The pass checked for a batch norm following the conv to avoid breaking fusion with a squeeze.
However, it did not support Conv -> Batch Norm -> ReLu OR Conv -> ReLU
This commit adds that support, along with other supported activation
Reviewed By: rascani
Differential Revision: D1050174691 parent afd32cc commit 215fa1b
5 files changed
Lines changed: 308 additions & 50 deletions
File tree
- backends/nxp
- aten_passes
- tests
- generic_tests
0 commit comments