From 45e4c1c4a260b063dd62404bdbd2089f91d668a9 Mon Sep 17 00:00:00 2001 From: Asim Shankar Date: Mon, 21 May 2018 19:30:52 -0700 Subject: s/tfe.GradientTape/tf.GradientTape/ s/tfe.enable_eager_execution/tf.enable_eager_execution/ PiperOrigin-RevId: 197500751 --- third_party/examples/eager/spinn/spinn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'third_party/examples') diff --git a/third_party/examples/eager/spinn/spinn.py b/third_party/examples/eager/spinn/spinn.py index 8a2b24aa4e..67456a5bdf 100644 --- a/third_party/examples/eager/spinn/spinn.py +++ b/third_party/examples/eager/spinn/spinn.py @@ -462,7 +462,7 @@ class SNLIClassifierTrainer(tfe.Checkpointable): 2. logits as a dense `Tensor` of shape (batch_size, d_out), where d_out is the output dimension size of the SNLIClassifier. """ - with tfe.GradientTape() as tape: + with tf.GradientTape() as tape: tape.watch(self._model.variables) logits = self._model(premise, premise_transition, -- cgit v1.2.3