Skip to content

Commit cf124e0

Browse files
jerryzh168facebook-github-bot
authored andcommitted
rename _quantize_pt2e to quantize_pt2e
Summary: X-link: pytorch/pytorch#105377 att Reviewed By: andrewor14 Differential Revision: D47234357 fbshipit-source-id: c5e433322f895b4753ffe283c63f07e8b7617019
1 parent 81ba4d4 commit cf124e0

File tree

2 files changed

+9
-8
lines changed

2 files changed

+9
-8
lines changed

backends/xnnpack/test/test_xnnpack_utils.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -41,13 +41,6 @@
4141
QConfigMapping,
4242
)
4343

44-
from torch.ao.quantization._quantize_pt2e import (
45-
convert_pt2e,
46-
get_symmetric_quantization_config,
47-
prepare_pt2e_quantizer,
48-
QNNPackQuantizer,
49-
)
50-
5144
from torch.ao.quantization.backend_config.executorch import (
5245
get_executorch_backend_config,
5346
)
@@ -67,6 +60,13 @@
6760
prepare_fx,
6861
)
6962

63+
from torch.ao.quantization.quantize_pt2e import (
64+
convert_pt2e,
65+
get_symmetric_quantization_config,
66+
prepare_pt2e_quantizer,
67+
QNNPackQuantizer,
68+
)
69+
7070
from torch.testing import FileCheck
7171

7272

exir/tests/test_quantization.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,15 +12,16 @@
1212
from torch.ao.ns.fx.utils import compute_sqnr
1313
from torch.ao.quantization import get_default_qconfig, QConfigMapping # @manual
1414

15-
from torch.ao.quantization._quantize_pt2e import convert_pt2e, prepare_pt2e_quantizer
1615
from torch.ao.quantization.backend_config._qnnpack_pt2e import (
1716
get_qnnpack_pt2e_backend_config,
1817
)
18+
1919
from torch.ao.quantization.pt2e.quantizer import QNNPackQuantizer
2020
from torch.ao.quantization.pt2e.quantizer.qnnpack_quantizer import (
2121
get_symmetric_quantization_config,
2222
)
2323
from torch.ao.quantization.quantize_fx import convert_to_reference_fx, prepare_fx
24+
from torch.ao.quantization.quantize_pt2e import convert_pt2e, prepare_pt2e_quantizer
2425
from torch.testing import FileCheck
2526
from torch.testing._internal.common_quantization import skipIfNoQNNPACK
2627
from torch.testing._internal.common_quantized import override_quantized_engine

0 commit comments

Comments
 (0)