|
|
|
|
|
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_fx.py`, while adding an import statement |
|
|
here. |
|
|
""" |
|
|
|
|
|
from torch.ao.quantization.quantize_fx import ( |
|
|
_check_is_graph_module, |
|
|
_swap_ff_with_fxff, |
|
|
_fuse_fx, |
|
|
Scope, |
|
|
ScopeContextManager, |
|
|
QuantizationTracer, |
|
|
_prepare_fx, |
|
|
_prepare_standalone_module_fx, |
|
|
fuse_fx, |
|
|
prepare_fx, |
|
|
prepare_qat_fx, |
|
|
_convert_fx, |
|
|
convert_fx, |
|
|
_convert_standalone_module_fx, |
|
|
) |
|
|
|
|
|
from torch.ao.quantization.fx.graph_module import ( |
|
|
ObservedGraphModule, |
|
|
) |
|
|
|
