Searched refs:rnn_type (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/python/keras/layers/ |
D | cudnn_recurrent_test.py | 266 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':
|