Commit 4474cba
Fix broken ConvBNReLu from new Convert1DConvTo2D pass (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 f3387d0 commit 4474cba
5 files changed
Lines changed: 305 additions & 51 deletions
File tree
- backends/nxp
- aten_passes
- tests
- generic_tests
0 commit comments