A CUDA kernel for NHWC GroupNorm for PyTorch
-
Updated
Nov 15, 2024 - Cuda
#
nhwc
Here is 1 public repository matching this topic...
Improve this page
Add a description, image, and links to the nhwc topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the nhwc topic, visit your repo's landing page and select "manage topics."