Calculate mean similarly to variance in _StreamMoments.
Calculate mean similarly to variance in _StreamMoments.

This collapses a few operators with ones in _Normalize.

PiperOrigin-RevId: 404299682

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo