pytorch

583ff947 - Fix max_pool2d for returning wrong shape with return_indices=True on cuda (#38992)

Fix max_pool2d for returning wrong shape with return_indices=True on cuda (#38992) Summary: Fix https://github.com/pytorch/pytorch/issues/38986 The current code only resizes pooling output but forget to resize indices as well. Pull Request resolved: https://github.com/pytorch/pytorch/pull/38992 Differential Revision: D21718324 Pulled By: ngimel fbshipit-source-id: 7cf937966d38ab2167be79979475c4e0cacbf82c