I see, thanks, I didn’t know that’s what cuDNN worked with that format.



 jekbradbury:

What you’re probably looking for is (something like) the code in the AutogradRNN class


I’m not exactly sure I’m correctly reading between the lines: are LSTMCell's inherently incapable of making use of cuDNN? Is what I’m trying to achieve simply a Bad Idea ™?