quantize_jit.py 713 B

1234567891011121314151617181920212223242526
  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_jit.py`, while adding an import statement
  7. here.
  8. """
  9. from torch.ao.quantization.quantize_jit import (
  10. _check_is_script_module,
  11. _check_forward_method,
  12. script_qconfig,
  13. script_qconfig_dict,
  14. fuse_conv_bn_jit,
  15. _prepare_jit,
  16. prepare_jit,
  17. prepare_dynamic_jit,
  18. _prepare_ondevice_dynamic_jit,
  19. _convert_jit,
  20. convert_jit,
  21. convert_dynamic_jit,
  22. _quantize_jit,
  23. quantize_jit,
  24. quantize_dynamic_jit
  25. )