Spaces:
Running
Running
| # 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, | |
| ) | |