Register assign ops to SpectralNormalization layer updates for TF1 compatibility.
Register assign ops to SpectralNormalization layer updates for TF1 compatibility.

This allow us to access all assignment ops via `layer.updates`. This is needed to enforce non-SGD-related variable updates in TF1 using `control_dependencies`.

PiperOrigin-RevId: 332535162

google/edward2