Fix a tensorflow deprecation warning by updating div to tf.math.divide.
Fix a tensorflow deprecation warning by updating div to tf.math.divide.

PiperOrigin-RevId: 328557519

google/deepvariant,google/deepvariant,google/deepvariant