ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-15 11:34:54 +00:00
Code Issues Packages Projects Releases Wiki Activity
2,678 Commits 779 Branches 40 Tags
c4fbbaaa5533b2689f4f3c42f0b3d5f3bb1376e5
Commit Graph

5 Commits

Author SHA1 Message Date
Mohsen Saffari
c4fbbaaa55 Add trait file for update_moving_average and save_mean_inv_std 2025-11-28 10:33:58 +00:00
Mohsen Saffari
b732595d9f add kSaveMeanInvStd, kUpdateMovingAverage in Traits 2025-11-27 15:06:13 +00:00
Mohsen Saffari
5c8e8684ec add gamma, bias to the simple kernel 2025-11-27 13:32:18 +00:00
Mohsen Saffari
c7afd540d9 correct the batch normalization algorithm to calc reduce over NxHxW 2025-11-25 16:46:00 +00:00
Mohsen Saffari
5fa3ec2b37 Some refactoring for simple batchnorm kernel and example 2025-11-25 13:14:58 +00:00