xref: /aosp_15_r20/external/pytorch/torch/quantization/fx/quantization_patterns.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1# flake8: noqa: F401
2r"""
3This file is in the process of migration to `torch/ao/quantization`, and
4is kept here for compatibility while the migration process is ongoing.
5If you are adding a new entry/functionality, please, add it to the
6appropriate files under `torch/ao/quantization/fx/`, while adding an import statement
7here.
8"""
9from torch.ao.quantization.fx.quantize_handler import (
10    BatchNormQuantizeHandler,
11    BinaryOpQuantizeHandler,
12    CatQuantizeHandler,
13    ConvReluQuantizeHandler,
14    CopyNodeQuantizeHandler,
15    CustomModuleQuantizeHandler,
16    DefaultNodeQuantizeHandler,
17    EmbeddingQuantizeHandler,
18    FixedQParamsOpQuantizeHandler,
19    GeneralTensorShapeOpQuantizeHandler,
20    LinearReLUQuantizeHandler,
21    QuantizeHandler,
22    RNNDynamicQuantizeHandler,
23    StandaloneModuleQuantizeHandler,
24)
25
26
27QuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
28BinaryOpQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
29CatQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
30ConvReluQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
31LinearReLUQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
32BatchNormQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
33EmbeddingQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
34RNNDynamicQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
35DefaultNodeQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
36FixedQParamsOpQuantizeHandler.__module__ = (
37    "torch.ao.quantization.fx.quantization_patterns"
38)
39CopyNodeQuantizeHandler.__module__ = "torch.ao.quantization.fx.quantization_patterns"
40CustomModuleQuantizeHandler.__module__ = (
41    "torch.ao.quantization.fx.quantization_patterns"
42)
43GeneralTensorShapeOpQuantizeHandler.__module__ = (
44    "torch.ao.quantization.fx.quantization_patterns"
45)
46StandaloneModuleQuantizeHandler.__module__ = (
47    "torch.ao.quantization.fx.quantization_patterns"
48)
49