Skip to content

Allow SKIP_INPUT for RNNs #633

@SeanNaren

Description

@SeanNaren

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!

Metadata

Metadata

Assignees

Labels

featureA request for a proper, new feature.todoNot 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 module

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions