Memory layout for pooling ops
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/25374
Test Plan: Imported from OSS
Differential Revision: D17107577
Pulled By: jamesr66a
fbshipit-source-id: e40dacaddf5ee17e6483be9e9302d3afc1a708c7