qconfig.py 910 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/qconfig.py`, while adding an import statement
  7. here.
  8. """
  9. from torch.ao.quantization.qconfig import (
  10. _add_module_to_qconfig_obs_ctr,
  11. _assert_valid_qconfig,
  12. default_activation_only_qconfig,
  13. default_debug_qconfig,
  14. default_dynamic_qconfig,
  15. default_per_channel_qconfig,
  16. default_qat_qconfig,
  17. default_qat_qconfig_v2,
  18. default_qconfig,
  19. default_weight_only_qconfig,
  20. float16_dynamic_qconfig,
  21. float16_static_qconfig,
  22. float_qparams_weight_only_qconfig,
  23. get_default_qat_qconfig,
  24. get_default_qconfig,
  25. per_channel_dynamic_qconfig,
  26. QConfig,
  27. qconfig_equals,
  28. QConfigAny,
  29. QConfigDynamic,
  30. )