xref: /aosp_15_r20/external/pytorch/torch/quantization/quantize_jit.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_jit.py`, while adding an import statement
7here.
8"""
9
10from torch.ao.quantization.quantize_jit import (
11    _check_forward_method,
12    _check_is_script_module,
13    _convert_jit,
14    _prepare_jit,
15    _prepare_ondevice_dynamic_jit,
16    _quantize_jit,
17    convert_dynamic_jit,
18    convert_jit,
19    fuse_conv_bn_jit,
20    prepare_dynamic_jit,
21    prepare_jit,
22    quantize_dynamic_jit,
23    quantize_jit,
24    script_qconfig,
25    script_qconfig_dict,
26)
27