Skip to content

Commit 5358bcf

Browse files
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: D105017469
1 parent bf438ce commit 5358bcf

5 files changed

Lines changed: 308 additions & 50 deletions

File tree

0 commit comments

Comments
 (0)