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

Update use of tf.CriticalSection.

Browse source 
PiperOrigin-RevId: 233168852
This commit is contained in:
Eugene Brevdo 2019-02-08 20:08:13 -08:00 committed by A. Unique TensorFlower
parent d75f1b80ba
commit 72305bcb10
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
privacy/analysis/privacy_ledger.py
View file

@ -76,6 +76,11 @@ class PrivacyLedger(object):
initial_value=0.0, trainable=False, name='sample_count') initial_value=0.0, trainable=False, name='sample_count')
self._query_count = tf.Variable( self._query_count = tf.Variable(
initial_value=0.0, trainable=False, name='query_count') initial_value=0.0, trainable=False, name='query_count')
try:
# Newer versions of TF
self._cs = tf.CriticalSection()
except AttributeError:
# Older versions of TF
self._cs = tf.contrib.framework.CriticalSection() self._cs = tf.contrib.framework.CriticalSection()
def record_sum_query(self, l2_norm_bound, noise_stddev): def record_sum_query(self, l2_norm_bound, noise_stddev):