def F32.sqdev

bend-tensors@0.0.0.2 · bend_tensors.bend:1983

import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensors
import Base

1983# layer norm over each row: (x - mean) / sqrt(var + eps), var the mean
1984# of the squared deviations; a pass for the sum, one for the squares,
1985# one to write
1986def F32.sqdev(+x: F32, +mean: F32) -> F32:
1987  ((x - mean) * (x - mean) : F32)

Uses (3)

Used by (1)