Skip to content

Commit 888f84d

Browse files
authored
update pt2e example code (#14284)
Summary: As title, moving to torchao Reviewed By: jerryzh168 Differential Revision: D82341472
1 parent 2faf5ce commit 888f84d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/source/backends-qualcomm.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -385,7 +385,7 @@ example_inputs = (torch.randn(1, 3, 224, 224),) # Example input tensor
385385
Choose between quantization approaches, post training quantization (PTQ) or quantization aware training (QAT):
386386
```python
387387
from executorch.backends.qualcomm.quantizer.quantizer import QnnQuantizer
388-
from torch.ao.quantization.quantize_pt2e import prepare_pt2e, prepare_qat_pt2e, convert_pt2e
388+
from torchao.quantization.pt2e.quantize_pt2e import prepare_pt2e, prepare_qat_pt2e, convert_pt2e
389389

390390
quantizer = QnnQuantizer()
391391
m = torch.export.export(model, example_inputs, strict=True).module()

0 commit comments

Comments
 (0)