Expand description
Normalisation requests for the cuDNN actor: BatchNorm (training + inference + persistent), LayerNorm, InstanceNorm, GroupNorm, RMSNorm.
Structs§
- Batch
Norm Request - BatchNorm — training-mode running-mean/var update + per-channel scale + bias.
- Group
Norm Request - GroupNorm — generalisation of BatchNorm/LayerNorm/InstanceNorm
with
groupschannel partitions. - Instance
Norm Request - InstanceNorm — normalises per-sample, per-channel.
- Layer
Norm Request - LayerNorm — normalises across the trailing axes, scale + bias applied per-feature.
- Norm
BwdRequest - Norm backward request, applies to BatchNorm / LayerNorm /
InstanceNorm / GroupNorm uniformly via
mode.
Functions§
- build_
norm_ fwd_ graph - Build the spec-side norm-fwd op graph.