rms_norm_forward_f64

void rms_norm_forward_f64(const double * x, const double * gamma, double * y, double * saved_rstd, int outer, int N, double eps)

Double-precision counterpart to rms_norm_forward_f32.

xconst double*

Input buffer of shape (outer, N).

gammaconst double*

Per-feature scale of shape (N,).

ydouble*

Output buffer of shape (outer, N).

saved_rstddouble*

Per-row reciprocal RMS, shape (outer,).

outer, Nstd::size_t

Layout extents.

epsdouble

Variance stabiliser.