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

tf_privacy.v1.make_vectorized_optimizer_class

Given a subclass of tf.compat.v1.train.Optimizer, returns a vectorized DP-SGD subclass of it.

cls Class from which to derive a DP subclass. Should be a subclass of tf.compat.v1.train.Optimizer.

A DP-SGD subclass of cls.