Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Feature request: CUDNN support for Dropout #12976

Closed
eric-haibin-lin opened this issue Oct 25, 2018 · 1 comment
Closed

Feature request: CUDNN support for Dropout #12976

eric-haibin-lin opened this issue Oct 25, 2018 · 1 comment
Labels
CUDA Feature request Operator

Comments

@eric-haibin-lin
Copy link
Member

Currently dropout operator doesn't have cudnn support. Furthermore, CUDNN doesn't support dropout with axis, while MXNet Naive dropout op does. I assume CUDNN dropout op will be much faster than MXNet's on large arrays.

@DickJC123 @ptrendx
@thomelane thomelane mentioned this issue Jan 10, 2019
Closed
@eric-haibin-lin
Copy link
Member Author

#13896

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Assignees
No one assigned
Labels
CUDA Feature request Operator
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@eric-haibin-lin