Implement aten::feature_dropout - #2404

Closed

Justin Chu (justinchuby) with Copilot wants to merge 5 commits into

mainfrom

copilot/fix-2403

Commits on Jun 19, 2025

Initial plan for issue
Copilot
committed
3b4c5a3
Implement aten_feature_dropout with proper torch_op decorator

Copilot
and
Justin Chu (justinchuby)
committed
658f966
Refactor aten_feature_dropout to use op.Dropout for cleaner implementation

Copilot
and
Justin Chu (justinchuby)
committed
29f1968
Add early return for p==0 or not train in aten_feature_dropout

Copilot
and
Justin Chu (justinchuby)
committed
af8d622
Refactor feature_dropout to use op.Shape with start/end parameters instead of op.Gather

Copilot
and
Justin Chu (justinchuby)
committed
96f78cd