tf_privacy.make_keras_optimizer_class

Given a subclass of tf.keras.optimizers.Optimizer, returns a DP-SGD subclass of it.

tf_privacy.make_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 DP-SGD subclass of cls.