pytorch

67a19fec - CUDA BFloat16 pooling (#45151)

CUDA BFloat16 pooling (#45151) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/45151 Reviewed By: ailzhang Differential Revision: D23854056 Pulled By: ngimel fbshipit-source-id: 32f0835218c2602a09654a9ac2d161c4eb360f90