From 789a05df63db4a2f31bd2821aefb954ab7fa45f9 Mon Sep 17 00:00:00 2001 From: Galen Andrew Date: Tue, 31 Aug 2021 08:41:23 -0700 Subject: [PATCH] Add DpEvent to TFP public symbols. PiperOrigin-RevId: 394010603 --- tensorflow_privacy/__init__.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/tensorflow_privacy/__init__.py b/tensorflow_privacy/__init__.py index 43426b6..a2e25d1 100644 --- a/tensorflow_privacy/__init__.py +++ b/tensorflow_privacy/__init__.py @@ -29,6 +29,20 @@ else: # TensorFlow v1 imports from tensorflow_privacy import v1 + # DpEvents + from tensorflow_privacy.privacy.analysis.dp_event import DpEvent + from tensorflow_privacy.privacy.analysis.dp_event import NoOpDpEvent + from tensorflow_privacy.privacy.analysis.dp_event import NonPrivateDpEvent + from tensorflow_privacy.privacy.analysis.dp_event import UnsupportedDpEvent + from tensorflow_privacy.privacy.analysis.dp_event import GaussianDpEvent + from tensorflow_privacy.privacy.analysis.dp_event import SelfComposedDpEvent + from tensorflow_privacy.privacy.analysis.dp_event import ComposedDpEvent + from tensorflow_privacy.privacy.analysis.dp_event import PoissonSampledDpEvent + from tensorflow_privacy.privacy.analysis.dp_event import FixedBatchSampledWrDpEvent + from tensorflow_privacy.privacy.analysis.dp_event import FixedBatchSampledWorDpEvent + from tensorflow_privacy.privacy.analysis.dp_event import ShuffledDatasetDpEvent + from tensorflow_privacy.privacy.analysis.dp_event import TreeAggregationDpEvent + # Analysis from tensorflow_privacy.privacy.analysis.compute_dp_sgd_privacy_lib import compute_dp_sgd_privacy from tensorflow_privacy.privacy.analysis.rdp_accountant import compute_heterogeneous_rdp