diff --git a/backends/xnnpack/test/test_xnnpack_utils.py b/backends/xnnpack/test/test_xnnpack_utils.py index acaedbf8cd7..7d261ee5544 100644 --- a/backends/xnnpack/test/test_xnnpack_utils.py +++ b/backends/xnnpack/test/test_xnnpack_utils.py @@ -42,11 +42,14 @@ QConfig, QConfigMapping, ) -from torch.ao.quantization._pt2e.quantizer import QNNPackQuantizer -from torch.ao.quantization._pt2e.quantizer.qnnpack_quantizer import ( + +from torch.ao.quantization._quantize_pt2e import ( + convert_pt2e, get_symmetric_quantization_config, + prepare_pt2e_quantizer, + QNNPackQuantizer, ) -from torch.ao.quantization._quantize_pt2e import convert_pt2e, prepare_pt2e_quantizer + from torch.ao.quantization.backend_config.executorch import ( get_executorch_backend_config, )