WILL BE REMOVED. DO NOT SUBMIT.

PiperOrigin-RevId: 562848204

Author: tensorflower-gardener

Diff for: tensorflow_privacy/privacy/dp_query/normalized_query.py

@@ -73,7 +73,7 @@ def get_noised_result(self, sample_state, global_state):
         sample_state, global_state.numerator_state)
 
     def normalize(v):
-      return tf.truediv(v, global_state.denominator)
+      return tf.truediv(v, tf.cast(global_state.denominator, v.dtype))
 
     # The denominator is constant so the privacy cost comes from the numerator.
     return (tf.nest.map_structure(normalize, noised_sum),