tf_privacy.make_keras_optimizer_class

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

cls Class from which to derive a DP subclass. Should be a subclass of tf.keras.optimizers.Optimizer.

A DP-SGD subclass of cls.