-
Notifications
You must be signed in to change notification settings - Fork 26.3k
Closed
Labels
featureA request for a proper, new feature.A request for a proper, new feature.todoNot as important as medium or high priority tasks, but we will work on these.Not as important as medium or high priority tasks, but we will work on these.triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
Description
cuDNN supports skipping the matrix multiplication with the incoming input via the CUDNN_SKIP_INPUT flag on the first recurrent layer. This was useful for passing in a pre-batchnormed input to the RNN.
Currently it's hardcoded to linear here and I assume this is due to being able to seamlessly move between torch RNNs and cudnn RNNs.
Would I be correct in thinking that just these cells need to be updated to take into consideration the flag? If so I'll get to work!
apaszke
Metadata
Metadata
Assignees
Labels
featureA request for a proper, new feature.A request for a proper, new feature.todoNot as important as medium or high priority tasks, but we will work on these.Not as important as medium or high priority tasks, but we will work on these.triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module