From d5e34b77c80e336ecc6009d72878bbe17b82329f Mon Sep 17 00:00:00 2001 From: Thomas O'Malley Date: Wed, 29 Jul 2020 16:44:00 -0700 Subject: [PATCH] Allow loss to be passed as a Tensor to Optimizer.minimize PiperOrigin-RevId: 323895500 --- tensorflow_privacy/privacy/bolt_on/optimizers_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_privacy/privacy/bolt_on/optimizers_test.py b/tensorflow_privacy/privacy/bolt_on/optimizers_test.py index 01341a7..b7227cf 100644 --- a/tensorflow_privacy/privacy/bolt_on/optimizers_test.py +++ b/tensorflow_privacy/privacy/bolt_on/optimizers_test.py @@ -151,7 +151,7 @@ class TestOptimizer(OptimizerV2): self.iterations = tf.constant(1, dtype=tf.float32) self._iterations = tf.constant(1, dtype=tf.float32) - def _compute_gradients(self, loss, var_list, grad_loss=None): + def _compute_gradients(self, loss, var_list, grad_loss=None, tape=None): return 'test' def get_config(self):