diff options
| author |  Asim Shankar <ashankar@google.com> | 2018-05-21 19:30:52 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-21 19:33:45 -0700 |
| commit | 45e4c1c4a260b063dd62404bdbd2089f91d668a9 (patch) | |
| tree | a8c69709050d09027d01e0df2e267b2c7b812f61 /third_party/examples | |
| parent | e4dcf28ad1c56c3a8e41ca52e7d87169eb7f93d5 (diff) | |
s/tfe.GradientTape/tf.GradientTape/
s/tfe.enable_eager_execution/tf.enable_eager_execution/
PiperOrigin-RevId: 197500751
Diffstat (limited to 'third_party/examples')
| -rw-r--r-- | third_party/examples/eager/spinn/spinn.py | 2 |
1 files changed, 1 insertions, 1 deletions
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, |