Use top-level torchao pt2e imports

Differential Revision: D75186690

Pull Request resolved: #11060

Author: metascroy

backends/example/example_quantizer.py

@@ -10,8 +10,11 @@
 import torch
 from executorch.backends.example.example_operators.ops import module_to_annotator
 from torch import fx
-from torchao.quantization.pt2e.graph_utils import find_sequential_partitions
-from torchao.quantization.pt2e.observer import HistogramObserver, MinMaxObserver
+from torchao.quantization.pt2e import (
+    find_sequential_partitions,
+    HistogramObserver,
+    MinMaxObserver,
+)
 from torchao.quantization.pt2e.quantizer import (
     OperatorConfig,
     QuantizationSpec,

backends/mediatek/quantizer/qconfig.py

@@ -10,8 +10,11 @@
 
 import torch
 
-from torchao.quantization.pt2e.fake_quantize import FakeQuantize
-from torchao.quantization.pt2e.observer import MinMaxObserver, PerChannelMinMaxObserver
+from torchao.quantization.pt2e import (
+    FakeQuantize,
+    MinMaxObserver,
+    PerChannelMinMaxObserver,
+)
 from torchao.quantization.pt2e.quantizer import QuantizationSpec
 
 

backends/nxp/quantizer/neutron_quantizer.py

@@ -28,16 +28,17 @@
     is_annotated,
     no_outside_users,
 )
-from executorch.backends.xnnpack.quantizer.xnnpack_quantizer_utils import (
+from torch import fx
+from torchao.quantization.pt2e import HistogramObserver, MinMaxObserver
+from torchao.quantization.pt2e.quantizer import (
+    ComposableQuantizer,
+    DerivedQuantizationSpec,
     OperatorConfig,
     QuantizationAnnotation,
     QuantizationConfig,
     QuantizationSpec,
+    Quantizer,
 )
-from torch import fx
-from torchao.quantization.pt2e.observer import HistogramObserver, MinMaxObserver
-from torchao.quantization.pt2e.quantizer import DerivedQuantizationSpec, Quantizer
-from torchao.quantization.pt2e.quantizer.composable_quantizer import ComposableQuantizer
 
 
 class NeutronAtenQuantizer(Quantizer):

backends/openvino/quantizer/quantizer.py

@@ -17,7 +17,7 @@
 import torch.fx
 
 from nncf.common.graph.graph import NNCFGraph  # type: ignore[import-untyped]
-from torchao.quantization.pt2e.observer import (
+from torchao.quantization.pt2e import (
     HistogramObserver,
     PerChannelMinMaxObserver,
     UniformQuantizationObserverBase,