Commit c81830af authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Don't cache RNN weights in while loops since it's possible that train steps... 

Don't cache RNN weights in while loops since it's possible that train steps could be updating the weights. This is specifically true on TPUS (tpu.repeat). Also, fix the `testDynamicRnnTrainLoop` unit test.

PiperOrigin-RevId: 202565323
parent 2273cda3
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