pytorch/aten
Danial Javady ed0e63e938 Add NHWC support for group normalization (#126635)
Fixes #111824

Currently it is the case that if the user specifies their group normalization to be of NHWC format, pytorch will default to NCHW tensors and convert. This  conversion is not immediately obvious to the user unless they check the format themselves which is not intuitive. This PR adds suppor for NHWC for cuda by adding necessary kernels.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/126635
Approved by: https://github.com/eqy, https://github.com/mikaylagawarecki
2024-11-07 01:12:08 +00:00
..
conda
src Add NHWC support for group normalization (#126635) 2024-11-07 01:12:08 +00:00
tools [BE][CI] Get rid of duplicated code (#131406) 2024-07-23 04:01:13 +00:00
CMakeLists.txt [Intel GPU] oneDNN GPU GEMM support (#117202) 2024-04-17 23:06:38 +00:00