pytorch/torch/ao
andrewor14 91190d8087 [quant][pt2e] Relax model_is_exported input (#120720)
Summary: This commit relaxes the `model_is_exported` API to
additionally work for `torch.nn.Module`s in addition to just
`torch.fx.GraphModule`s, simplifying downstream uses.

Test Plan:
python test/test_quantization.py TestQuantizePT2E.test_model_is_exported

Differential Revision: [D54263935](https://our.internmc.facebook.com/intern/diff/D54263935)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/120720
Approved by: https://github.com/tugsbayasgalan
2024-02-28 18:32:03 +00:00
..
nn make internal lintrunner mypy clean (#119840) 2024-02-14 00:25:42 +00:00
ns Enable possibly-undefined error code (#118533) 2024-01-30 21:07:01 +00:00
pruning Enable possibly-undefined error code (#118533) 2024-01-30 21:07:01 +00:00
quantization [quant][pt2e] Relax model_is_exported input (#120720) 2024-02-28 18:32:03 +00:00
__init__.py