| 123456789101112131415161718192021222324252627282930 |
- # flake8: noqa: F401
- r"""
- This file is in the process of migration to `torch/ao/quantization`, and
- is kept here for compatibility while the migration process is ongoing.
- If you are adding a new entry/functionality, please, add it to the
- `torch/ao/quantization/quantize.py`, while adding an import statement
- here.
- """
- from torch.ao.quantization.quantize import (
- _add_observer_,
- _convert,
- _get_observer_dict,
- _get_unique_devices_,
- _is_activation_post_process,
- _observer_forward_hook,
- _propagate_qconfig_helper,
- _register_activation_post_process_hook,
- _remove_activation_post_process,
- _remove_qconfig,
- add_quant_dequant,
- convert,
- prepare,
- prepare_qat,
- propagate_qconfig_,
- quantize,
- quantize_dynamic,
- quantize_qat,
- swap_module,
- )
|