Commit 83df61b4 authored by Scott Zhu's avatar Scott Zhu Committed by TensorFlower Gardener
Browse files

Fix the input_spec for RNN when stateful=True. 

The inputs spec was populated by RNN.build() but was overwritten when initial_state was provided during call(). We will need the populated input spec since it might be used when stateful=True.

See more details in #25985

PiperOrigin-RevId: 237610839
parent 5d43d943
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