ruby/tensorflow_privacy
1
0
Fork 0
forked from 626_privacy/tensorflow_privacy
Code Pull requests Activity

Ensures types remain consistent.

Browse source 
PiperOrigin-RevId: 563244784
This commit is contained in:
A. Unique TensorFlower 2023-09-06 16:12:46 -07:00
parent c92610e37a
commit a23cccde8b
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tensorflow_privacy/privacy/dp_query/normalized_query.py
View file

@ -73,7 +73,7 @@ class NormalizedQuery(dp_query.SumAggregationDPQuery):
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),