Add DpEvent to TFP public symbols.

PiperOrigin-RevId: 394010603
This commit is contained in:
Galen Andrew 2021-08-31 08:41:23 -07:00 committed by A. Unique TensorFlower
parent 3059fbae67
commit 789a05df63

View file

@ -29,6 +29,20 @@ else:
# TensorFlow v1 imports # TensorFlow v1 imports
from tensorflow_privacy import v1 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 # Analysis
from tensorflow_privacy.privacy.analysis.compute_dp_sgd_privacy_lib import compute_dp_sgd_privacy 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 from tensorflow_privacy.privacy.analysis.rdp_accountant import compute_heterogeneous_rdp