mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
44e3ed88c9
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/64086 AO Team is migrating the existing torch.quantization into torch.ao.quantization. We are doing it one file at a time to make sure that the internal callsites are updated properly. This migrates the `quantize.py` from torch.quantization to `torch.ao.quantization`. At this point both locations will be supported. Eventually the torch.quantization will be deprecated. Test Plan: `buck test mode/opt //caffe2/test:quantization` Reviewed By: jerryzh168, raghuramank100 Differential Revision: D30055886 fbshipit-source-id: 8ef7470f9fa640c0042bef5bb843e7a05ecd0b9f
31 lines
1.5 KiB
Python
31 lines
1.5 KiB
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.
|
|
"""
|
|
|
|
import torch
|
|
|
|
from torch.ao.quantization.quantize import _convert
|
|
from torch.ao.quantization.quantize import _observer_forward_hook
|
|
from torch.ao.quantization.quantize import _propagate_qconfig_helper
|
|
from torch.ao.quantization.quantize import _remove_activation_post_process
|
|
from torch.ao.quantization.quantize import _remove_qconfig
|
|
from torch.ao.quantization.quantize import add_observer_
|
|
from torch.ao.quantization.quantize import add_quant_dequant
|
|
from torch.ao.quantization.quantize import convert
|
|
from torch.ao.quantization.quantize import get_observer_dict
|
|
from torch.ao.quantization.quantize import get_unique_devices_
|
|
from torch.ao.quantization.quantize import is_activation_post_process
|
|
from torch.ao.quantization.quantize import prepare
|
|
from torch.ao.quantization.quantize import prepare_qat
|
|
from torch.ao.quantization.quantize import propagate_qconfig_
|
|
from torch.ao.quantization.quantize import quantize
|
|
from torch.ao.quantization.quantize import quantize_dynamic
|
|
from torch.ao.quantization.quantize import quantize_qat
|
|
from torch.ao.quantization.quantize import register_activation_post_process_hook
|
|
from torch.ao.quantization.quantize import swap_module
|