mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
Export team is fixing up the old strict export implementation, as a result it fails a check where we proxy the whole module under given directories. _WrapperModule is a way for torchao to workaround the issue where export requiring nn.module to trace so it should never get proxied in the graph. Differential Revision: [D82732613](https://our.internmc.facebook.com/intern/diff/D82732613) Pull Request resolved: https://github.com/pytorch/pytorch/pull/163258 Approved by: https://github.com/anijain2305 ghstack dependencies: #163136, #163137 |
||
|---|---|---|
| .. | ||
| nn | ||
| ns | ||
| pruning | ||
| quantization | ||
| __init__.py | ||