Commit 93f2998a authored by Allen Lavoie's avatar Allen Lavoie Committed by TensorFlower Gardener
Browse files

Make CUDNN RNN compatible with eager execution's kernel caching. 

Allows multiple CUDNN RNN calls with different shapes to share the same kernel.

Adds an input_shape-keyed scratch space cache to the kernel. This also fixes shape errors when a CUDNN RNN kernel was presented with multiple shapes during graph execution (e.g. from a while_loop).

Fixes #15752.

PiperOrigin-RevId: 180998667
parent 6ed75e60
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