Migrate some missed pt2e pieces (#11500)

Author: metascroy

backends/arm/_passes/scalars_to_attribute_pass.py

@@ -12,8 +12,8 @@
 from executorch.backends.arm._passes.arm_pass_utils import get_first_fake_tensor
 
 from executorch.exir.pass_base import ExportPass, PassResult
-from torch.ao.quantization.fx.utils import get_new_attr_name_with_prefix
 from torch.fx import GraphModule, Node
+from torchao.quantization.pt2e.utils import get_new_attr_name_with_prefix
 
 
 class ScalarsToAttributePass(ExportPass):

backends/qualcomm/_passes/lift_constant_scalar_operands.py

@@ -14,8 +14,8 @@
 from executorch.exir.pass_base import ExportPass, PassResult
 from executorch.exir.passes import dead_code_elimination_pass
 from torch import fx
-from torch.ao.quantization.fx.utils import get_new_attr_name_with_prefix
 from torch.ops import aten as aten
+from torchao.quantization.pt2e.utils import get_new_attr_name_with_prefix
 
 
 @dataclass(frozen=True)

export/export.py

@@ -15,11 +15,12 @@
 from executorch.runtime import Runtime, Verification
 from tabulate import tabulate
 from torch import nn
-from torch.ao.quantization import allow_exported_model_train_eval
-from torch.ao.quantization.quantizer.composable_quantizer import ComposableQuantizer
 from torch.export import ExportedProgram
 from torchao.quantization import quantize_
+from torchao.quantization.pt2e import allow_exported_model_train_eval
 from torchao.quantization.pt2e.quantize_pt2e import convert_pt2e, prepare_pt2e
+
+from torchao.quantization.pt2e.quantizer import ComposableQuantizer
 from torchao.utils import unwrap_tensor_subclass
 
 from .recipe import ExportRecipe