Skip to content

Commit 4474cba

Browse files
JakeStevensfacebook-github-bot
authored andcommitted
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: D105017469
1 parent f3387d0 commit 4474cba

5 files changed

Lines changed: 305 additions & 51 deletions

File tree

0 commit comments

Comments
 (0)