Add CuDNN GRU and LSTM layers. (keras-team#8094)

* temp version of CuDNNGRU * Add CuDNN GRU and LSTM layers. * Restrict tests to GPU configs. * Fix test skipping * Fix test flake with CNTK.
deltheil · Oct 9, 2017 · b3370c0 · b3370c0
1 parent eac0d42
commit b3370c0
Show file tree

Hide file tree

Showing 4 changed files with 877 additions and 0 deletions.
diff --git a/keras/layers/__init__.py b/keras/layers/__init__.py
@@ -11,6 +11,7 @@
 from .pooling import *
 from .local import *
 from .recurrent import *
+from .cudnn_recurrent import *
 from .normalization import *
 from .embeddings import *
 from .noise import *