diff options
| author |  Allen Lavoie <allenl@google.com> | 2018-05-10 18:28:24 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-10 18:30:54 -0700 |
| commit | 56b46370ba08c76200711f4a8d25194af1235fd5 (patch) | |
| tree | e9d4a9822884671935a2047059624eaab5d0580a /third_party | |
| parent | 5cef54072782a9a893eda69bec30fcf79cd0086b (diff) | |
Checkpointable: Have RNN wrappers add their cells as dependencies
Also marks _SlimRNNCell as not checkpointable, and adds a more convenient way to tag such classes.
Ideally adding a wrapper around a cell wouldn't break a checkpoint. This could look like RNN cell wrappers inheriting the dependencies of the cell they're wrapping. Possible to add that later if there's demand, or users can just add a dependency on wrapper._cell in addition to/instead of the wrapper when modifying programs.
Fixes #19208.
PiperOrigin-RevId: 196202366
Diffstat (limited to 'third_party')
0 files changed, 0 insertions, 0 deletions