31 lines
804 B
Python
31 lines
804 B
Python
# 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,
|
|
)
|