1234567891011121314151617181920212223242526 |
- # 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_jit.py`, while adding an import statement
- here.
- """
- from torch.ao.quantization.quantize_jit import (
- _check_is_script_module,
- _check_forward_method,
- script_qconfig,
- script_qconfig_dict,
- fuse_conv_bn_jit,
- _prepare_jit,
- prepare_jit,
- prepare_dynamic_jit,
- _prepare_ondevice_dynamic_jit,
- _convert_jit,
- convert_jit,
- convert_dynamic_jit,
- _quantize_jit,
- quantize_jit,
- quantize_dynamic_jit
- )
|