frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[functorch] Added dropout (tested by opinfo to be upstreamed)

Browse Source
This commit is contained in:
Horace He
2021-06-30 12:39:42 -07:00
committed by Jon Janzen
parent a598ed6be6
commit 0bba49c8a7
2 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
functorch/functorch/csrc/BatchRulesConv.cpp → functorch/functorch/csrc/BatchRulesModules.cpp
View File

@ -209,5 +209,6 @@ TORCH_LIBRARY_IMPL(aten, FT_BATCHED_KEY, m) {
m.impl("mkldnn_convolution", mkldnn_convolution_decomp);
m.impl("cudnn_convolution_backward", cudnn_convolution_backward_plumbing);
m.impl("cudnn_convolution", cudnn_convolution_plumbing);
OP_DECOMPOSE(dropout);
}
}}

1
functorch/test/functorch_lagging_op_db.py
View File

@ -210,6 +210,7 @@ _functorch_lagging_meta = {
('narrow', ''),
('ne', ''),
('neg', ''),
('nn.functional.dropout', ''),
('nn.functional.gelu', ''),
('nn.functional.hardshrink', ''),
('nn.functional.hardswish', ''),