From 78ec3fa58a1fdeec0311dfa2096ddfdb91e1aaa5 Mon Sep 17 00:00:00 2001 From: pranav subramani Date: Fri, 8 Jan 2021 00:24:52 -0700 Subject: [PATCH] update dp keras model --- tensorflow_privacy/privacy/keras_models/dp_keras_model.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tensorflow_privacy/privacy/keras_models/dp_keras_model.py b/tensorflow_privacy/privacy/keras_models/dp_keras_model.py index b854f54..19b6fd9 100644 --- a/tensorflow_privacy/privacy/keras_models/dp_keras_model.py +++ b/tensorflow_privacy/privacy/keras_models/dp_keras_model.py @@ -1,3 +1,7 @@ +''' +Keras Model for vectorized dpsgd with XLA acceleration +''' + import tensorflow as tf