clab/dynet

Add cuDNN support for RNNs

Open

#574 aperta il 31 mag 2017

Vedi su GitHub
 (0 commenti) (0 reazioni) (0 assegnatari)C++ (701 fork)github user discovery
enhancementhelp wanted

Metriche repository

Star
 (3433 star)
Metriche merge PR
 (Metriche PR in attesa)

Descrizione

It would be nice to have cuDNN speedup for SimpleRNN, LSTM and GRU.

While cuDNN will likely not be able to accelerate incremental RNN construction, we can special case the .transduce(seq) and .add_inputs(seq) implementations to make use of the acceleration.

Guida contributor