diff options
| author |  Tom Hennigan <tomhennigan@google.com> | 2018-09-07 02:56:32 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-07 03:00:56 -0700 |
| commit | c35e6928dc0b00ca3ca489e5d34f856499cece4a (patch) | |
| tree | 799f8dffe4ea6bf096dcd4ecc178bcf34f4ed19f /tensorflow/tools/api | |
| parent | 5a635e3472e16007830fca533c35b2f63fc4f898 (diff) | |
Support not automatically watching (trainable) accessed variables in GradientTape.
For more complex use cases this allows fine grained control over what is tracked
by the tape.
PiperOrigin-RevId: 211948236
Diffstat (limited to 'tensorflow/tools/api')
| -rw-r--r-- | tensorflow/tools/api/golden/v1/tensorflow.-gradient-tape.pbtxt | 2 | ||||
| -rw-r--r-- | tensorflow/tools/api/golden/v2/tensorflow.-gradient-tape.pbtxt | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/tools/api/golden/v1/tensorflow.-gradient-tape.pbtxt b/tensorflow/tools/api/golden/v1/tensorflow.-gradient-tape.pbtxt index cbf655498c..2f4257a66a 100644 --- a/tensorflow/tools/api/golden/v1/tensorflow.-gradient-tape.pbtxt +++ b/tensorflow/tools/api/golden/v1/tensorflow.-gradient-tape.pbtxt @@ -4,7 +4,7 @@ tf_class { is_instance: "<type \'object\'>" member_method { name: "__init__" - argspec: "args=[\'self\', \'persistent\'], varargs=None, keywords=None, defaults=[\'False\'], " + argspec: "args=[\'self\', \'persistent\', \'watch_accessed_variables\'], varargs=None, keywords=None, defaults=[\'False\', \'True\'], " } member_method { name: "gradient" diff --git a/tensorflow/tools/api/golden/v2/tensorflow.-gradient-tape.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.-gradient-tape.pbtxt index cbf655498c..2f4257a66a 100644 --- a/tensorflow/tools/api/golden/v2/tensorflow.-gradient-tape.pbtxt +++ b/tensorflow/tools/api/golden/v2/tensorflow.-gradient-tape.pbtxt @@ -4,7 +4,7 @@ tf_class { is_instance: "<type \'object\'>" member_method { name: "__init__" - argspec: "args=[\'self\', \'persistent\'], varargs=None, keywords=None, defaults=[\'False\'], " + argspec: "args=[\'self\', \'persistent\', \'watch_accessed_variables\'], varargs=None, keywords=None, defaults=[\'False\', \'True\'], " } member_method { name: "gradient" |