Commit 5358bcf
authored
Fix broken ConvBNReLu from new Convert1DConvTo2D pass (pytorch#19558) (pytorch#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 bf438ce commit 5358bcf
5 files changed
Lines changed: 308 additions & 50 deletions
File tree
- backends/nxp
- aten_passes
- tests
- generic_tests
0 commit comments