Commit 36304bc4 authored by Scott Zhu's avatar Scott Zhu Committed by TensorFlower Gardener
Browse files

Add dtype check for RNN cells' inputs.

For Basic RNN, GRU and LSTM, those cells use tanh and sigmoid activation
function, which only support floats and complex as input types.

See #23878 for more context.

PiperOrigin-RevId: 225890009
parent 282329c5
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment