Update seq2seq test to be 2.0 compatible.
Previously it was only tested under eager mode, and new issue was discovered by when "TF2_BEHAVIOR=1". Also the v2 tests has been updated to use keras.LSTMCell to ensure the correctness in v2. PiperOrigin-RevId: 236470595
Loading
Please sign in to comment