Include max pool in fusion groups (#52613)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/52613
Including MaxPool as part of the MKLDNN fusion group sped up resnet18 by ~20%, and was a win on other models I tested as well. I will post more complete benchmarks.
As mentioned in the diff, in some cases MaxPool can be slower than aten - ideally we'd only include maxpool if it decreased the number of layout transformations that occur. That hasnt actually matttered for all of the torchvision models, I don't think its necessary for this PR.
Test Plan: Imported from OSS
Reviewed By: navahgar
Differential Revision: D26696704
Pulled By: eellison
fbshipit-source-id: 61a025dbf5e7591c0a0f75def3beb439a138a21e