View source on GitHub |
Given a subclass of tf.keras.optimizers.Optimizer
, returns a vectorized DP-SGD subclass of it.
tf_privacy.make_vectorized_keras_optimizer_class(
cls
)
Args | |
---|---|
cls
|
Class from which to derive a DP subclass. Should be a subclass of
tf.keras.optimizers.Optimizer .
|
Returns | |
---|---|
A vectorized DP-SGD subclass of cls .
|