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
Loading
Please sign in to comment