Commit 87e56623 authored by Scott Zhu's avatar Scott Zhu Committed by TensorFlower Gardener
Browse files

[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
parent 4be575a6
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment