Remove tf.function annotation from quantile_adaptive_clip_sum_query.py that was causing import error.

PiperOrigin-RevId: 248236331
This commit is contained in:
Steve Chien 2019-05-14 16:40:35 -07:00 committed by A. Unique TensorFlower
parent aaf029edad
commit 17fefb3895

View file

@ -120,7 +120,6 @@ class QuantileAdaptiveClipSumQuery(dp_query.DPQuery):
self._sum_query.initial_global_state(),
self._clipped_fraction_query.initial_global_state())
@tf.function
def derive_sample_params(self, global_state):
"""See base class."""
gs = global_state