Fixed a serialization bug in the Bayesian initializers.
Fixed a serialization bug in the Bayesian initializers.

PiperOrigin-RevId: 235972599

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor