Function siluRmsHeads

siluRmsHeads(
    x: MlxArray,
    qScale: MlxArray,
    kScale: MlxArray,
    keyHeads: number,
    valueHeads: number,
    valueWidth: number,
): [MlxArray, MlxArray, MlxArray]
Convolved [B,S,2HkDk+Hv*Dv] to owned Q, K, V heads, Dk/Dv <=128. Scale vectors have Dk entries in the input dtype, as MLX's RMS weights do.
Parameters
- x: MlxArray
- qScale: MlxArray
- kScale: MlxArray
- keyHeads: number
- valueHeads: number
- valueWidth: number
Returns [MlxArray, MlxArray, MlxArray]
- Defined in packages/inference/src/kernels/delta/silu-rms-heads.ts:49