ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf_privacy.make_vectorized_keras_optimizer_class

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

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

A vectorized DP-SGD subclass of cls.