xref: /aosp_15_r20/external/pytorch/torch/quantization/quantize.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.py`, while adding an import statement
7here.
8"""
9
10from torch.ao.quantization.quantize import (
11    _add_observer_,
12    _convert,
13    _get_observer_dict,
14    _get_unique_devices_,
15    _is_activation_post_process,
16    _observer_forward_hook,
17    _propagate_qconfig_helper,
18    _register_activation_post_process_hook,
19    _remove_activation_post_process,
20    _remove_qconfig,
21    add_quant_dequant,
22    convert,
23    prepare,
24    prepare_qat,
25    propagate_qconfig_,
26    quantize,
27    quantize_dynamic,
28    quantize_qat,
29    swap_module,
30)
31