quantize.py 804 B

123456789101112131415161718192021222324252627282930
  1. # flake8: noqa: F401
  2. r"""
  3. This file is in the process of migration to `torch/ao/quantization`, and
  4. is kept here for compatibility while the migration process is ongoing.
  5. If you are adding a new entry/functionality, please, add it to the
  6. `torch/ao/quantization/quantize.py`, while adding an import statement
  7. here.
  8. """
  9. from torch.ao.quantization.quantize import (
  10. _add_observer_,
  11. _convert,
  12. _get_observer_dict,
  13. _get_unique_devices_,
  14. _is_activation_post_process,
  15. _observer_forward_hook,
  16. _propagate_qconfig_helper,
  17. _register_activation_post_process_hook,
  18. _remove_activation_post_process,
  19. _remove_qconfig,
  20. add_quant_dequant,
  21. convert,
  22. prepare,
  23. prepare_qat,
  24. propagate_qconfig_,
  25. quantize,
  26. quantize_dynamic,
  27. quantize_qat,
  28. swap_module,
  29. )