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