Home
last modified time | relevance | path

Searched refs:rnn_type (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/keras/layers/
Dcudnn_recurrent_test.py266 rnn_type=['LSTM', 'GRU'], to_cudnn=[True, False],
271 def test_load_weights_between_noncudnn_rnn(self, rnn_type, to_cudnn, argument
287 if rnn_type == 'LSTM':
347 rnn_type=['LSTM', 'GRU'], to_cudnn=[True, False]))
350 def test_load_weights_between_noncudnn_rnn_time_distributed(self, rnn_type, argument
367 if rnn_type == 'LSTM':