Skip to content

Testing the CUDA implementation of the LSTM cell with tensorflow bindings

Notifications You must be signed in to change notification settings

C-J-Cundy/CudnnLSTM-test

Repository files navigation

CudnnLSTM-test

Testing the CUDA implementation of the LSTM cell with tensorflow bindings. It was very hard to find worked examples of using the CUDNN hooks in tensorflow, so a full example on synthetic data is provided here, including user-specified initialization.

There is also assorted code for use in the PLR project with Eric Martin: paper available here

About

Testing the CUDA implementation of the LSTM cell with tensorflow bindings

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published