Update covariance reset method for GP Layer.
Update covariance reset method for GP Layer.

This change of reset covariance by assigning the default value to an existing `tf.Variable` rather than creating a new layer. The latter is not allowed in the graph mode.

PiperOrigin-RevId: 342324256

google/edward2