clab/dynet

Add cuDNN support for RNNs

Open

#574 aberto em 31 de mai. de 2017

Ver no GitHub
 (0 comments) (0 reactions) (0 assignees)C++ (701 forks)github user discovery
enhancementhelp wanted

Métricas do repositório

Stars
 (3.433 stars)
Métricas de merge de PR
 (Métricas PR pendentes)

Description

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.

Guia do colaborador