pytorch

9e7c4764 - [NHWC CUDNN CONV]Update cudnn convolution memory_format behavior (#32482)

[NHWC CUDNN CONV]Update cudnn convolution memory_format behavior (#32482) Summary: 1. Allows both the memory_format of weight & input to dictate the output memory_format. 2. Provides utility function to recursively convert memory_format of Conv2d and ConvTranspose2d layers. This allows easy model conversion and ensures that lost memory_format through incompatible layers could be restored at Convolution-like layer, where significant performance boost is expected on later generation CUDA devices. Pull Request resolved: https://github.com/pytorch/pytorch/pull/32482 Differential Revision: D19647903 Pulled By: VitalyFedyunin fbshipit-source-id: 62c96ff6208ff5e84fae1f55b63af9a010ad199a