diff options
| author |  Alexandre Passos <apassos@google.com> | 2018-05-17 08:23:10 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-17 08:25:40 -0700 |
| commit | 9a75743f7a4190c788a33ec7bd4b384e12292cb1 (patch) | |
| tree | 5a82790fb7b08a10df0aefa34d89a191e845d41d /tensorflow/tools | |
| parent | ec0659b06a07f010123039baa16d120bd0752dda (diff) | |
Methods to stop and reset tf.GradientTape()
PiperOrigin-RevId: 196995160
Diffstat (limited to 'tensorflow/tools')
| -rw-r--r-- | tensorflow/tools/api/golden/tensorflow.-gradient-tape.pbtxt | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/tools/api/golden/tensorflow.-gradient-tape.pbtxt b/tensorflow/tools/api/golden/tensorflow.-gradient-tape.pbtxt index 7405202b89..cbf655498c 100644 --- a/tensorflow/tools/api/golden/tensorflow.-gradient-tape.pbtxt +++ b/tensorflow/tools/api/golden/tensorflow.-gradient-tape.pbtxt @@ -11,6 +11,14 @@ tf_class { argspec: "args=[\'self\', \'target\', \'sources\', \'output_gradients\'], varargs=None, keywords=None, defaults=[\'None\'], " } member_method { + name: "reset" + argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "stop_recording" + argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None" + } + member_method { name: "watch" argspec: "args=[\'self\', \'tensor\'], varargs=None, keywords=None, defaults=None" } |