Commit a936b239 authored by James Qin's avatar James Qin Committed by TensorFlower Gardener
Browse files

Support reuse cuDNNLSTM-trained checkpoints by multi-layer LSTM(Block)Cell 

* Add tensor stitching/partition in RNNParamSaveable for saving/restoring properly shaped and formatted weights/biases to share w/ LSTM(Block)Cell
* Add remapped names for canonical tensors during saving.
* Unittests

PiperOrigin-RevId: 159634913
parent 4be28767
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment