cudnn_recurrent_test.py | 266 rnn_type=['LSTM', 'GRU'], to_cudnn=[True, False], 291 rnn_layer_class = keras.layers.GRU 347 rnn_type=['LSTM', 'GRU'], to_cudnn=[True, False])) 371 rnn_layer_class = keras.layers.GRU 455 def gru(cudnn=False, **kwargs): function in function:CuDNNV1OnlyTest.test_preprocess_weights_for_loading_gru_incompatible 456 layer_class = keras.layers.CuDNNGRU if cudnn else keras.layers.GRU 471 gru(), 472 gru(cudnn=True), 473 'GRU(reset_after=False) is not compatible with CuDNNGRU') 475 gru(cudnn=True) [all...] |