pytorch

283ce12a - Add channels_last3d support for mkldnn conv and mkldnn deconv (#95271)

Add channels_last3d support for mkldnn conv and mkldnn deconv (#95271) ### Motivation - Add channels_last3d support for mkldnn conv and mkldnn deconv. - Use `ideep::convolution_transpose_forward::compute_v3` instead of `ideep::convolution_transpose_forward::compute`. compute_v3 uses `is_channels_last` to notify ideep whether to go CL or not to align with the memory format check of PyTorch. ### Testing 1 socket (28 cores): - memory format: torch.contiguous_format module | shape | forward / ms | backward / ms -- | -- | -- | -- conv3d | input size: (32, 32, 10, 100, 100), weight size: (32, 32, 3, 3, 3) | 64.56885 | 150.1796 conv3d | input size: (32, 16, 10, 200, 200), weight size: (16, 16, 3, 3, 3) | 100.6754 | 231.8883 conv3d | input size: (16, 4, 5, 300, 300), weight size: (4, 4, 3, 3, 3) | 19.31751 | 68.31131 module | shape | forward / ms | backward / ms -- | -- | -- | -- ConvTranspose3d | input size: (32, 32, 10, 100, 100), weight size: (32, 32, 3, 3, 3) | 122.7646 | 207.5125 ConvTranspose3d | input size: (32, 16, 10, 200, 200), weight size: (16, 16, 3, 3, 3) | 202.4542 | 368.5492 ConvTranspose3d | input size: (16, 4, 5, 300, 300), weight size: (4, 4, 3, 3, 3) | 122.959 | 84.62577 - memory format: torch.channels_last_3d module | shape | forward / ms | backward / ms -- | -- | -- | -- conv3d | input size: (32, 32, 10, 100, 100), weight size: (32, 32, 3, 3, 3) | 40.06993 | 114.317 conv3d | input size: (32, 16, 10, 200, 200), weight size: (16, 16, 3, 3, 3 | 49.08249 | 133.4079 conv3d | input size: (16, 4, 5, 300, 300), weight size: (4, 4, 3, 3, 3) | 5.873911 | 17.58647 module | shape | forward / ms | backward / ms -- | -- | -- | -- ConvTranspose3d | input size: (32, 32, 10, 100, 100), weight size: (32, 32, 3, 3, 3) | 88.4246 | 208.2269 ConvTranspose3d | input size: (32, 16, 10, 200, 200), weight size: (16, 16, 3, 3, 3 | 140.0725 | 270.4172 ConvTranspose3d | input size: (16, 4, 5, 300, 300), weight size: (4, 4, 3, 3, 3) | 23.0223 | 37.16972 Pull Request resolved: https://github.com/pytorch/pytorch/pull/95271 Approved by: https://github.com/jgong5, https://github.com/cpuhrsch