WILL BE REMOVED. DO NOT SUBMIT.

PiperOrigin-RevId: 562832706
tensorflow · Sep 6, 2023 · a49369f · a49369f
1 parent c92610e
commit a49369f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_privacy/privacy/dp_query/normalized_query.py b/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),