Module group_normalization 

Actual channel-group normalization and selected first-order derivatives.

Structs

GroupNormGeometry

Actual logical channel-group geometry, shared by backend dispatch and operation graphs.

Functions

geometry

Validate actual [batch, channels, ...] input and evenly divided nonempty groups.

group_norm_backward_select

Independently requested input/actual-weight/bias derivatives on the same backend.

group_norm_with_stats

Same-device biased-variance GroupNorm with FP32/FP64 saved [batch, groups] statistics.