xref: /aosp_15_r20/external/pytorch/torch/quantization/quantize_fx.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
6`torch/ao/quantization/quantize_fx.py`, while adding an import statement
7here.
8"""
9
10from torch.ao.quantization.fx.graph_module import ObservedGraphModule
11from torch.ao.quantization.quantize_fx import (
12    _check_is_graph_module,
13    _convert_fx,
14    _convert_standalone_module_fx,
15    _fuse_fx,
16    _prepare_fx,
17    _prepare_standalone_module_fx,
18    _swap_ff_with_fxff,
19    convert_fx,
20    fuse_fx,
21    prepare_fx,
22    prepare_qat_fx,
23    QuantizationTracer,
24    Scope,
25    ScopeContextManager,
26)
27