pytorch
c886406c - Add dropout shape inference as no-op in acc_tracer (#65113)

Commit
3 years ago
Add dropout shape inference as no-op in acc_tracer (#65113) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/65113 Register dropout as no-op in acc_tracer & Add shape inference for no-op Test Plan: buck test glow/fb/fx/acc_tracer:test_acc_shape_inference --- test_unary_15_dropout_no_op buck test glow/fb/fx/oss_acc_tracer:test_acc_tracer -- test_dropout Reviewed By: jfix71 Differential Revision: D30880679 fbshipit-source-id: 592fe50e17137c94c12727658191dedf08daf8cf
Author
Parents
Loading