[RNN] Update TF GRU and BasicRNN cell to support Keras RNN layer.
Keras RNN layer will feed in state_t as a list, and expect state_t+1 as list as well, even the state is a single tensor for simple cell. On the other hand, TF RNN does not wrap single tensor state. PiperOrigin-RevId: 208666836
Loading
Please sign in to comment