quantize_fx.py 746 B

1234567891011121314151617181920212223242526272829
  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_fx.py`, while adding an import statement
  7. here.
  8. """
  9. from torch.ao.quantization.quantize_fx import (
  10. _check_is_graph_module,
  11. _swap_ff_with_fxff,
  12. _fuse_fx,
  13. Scope,
  14. ScopeContextManager,
  15. QuantizationTracer,
  16. _prepare_fx,
  17. _prepare_standalone_module_fx,
  18. fuse_fx,
  19. prepare_fx,
  20. prepare_qat_fx,
  21. _convert_fx,
  22. convert_fx,
  23. _convert_standalone_module_fx,
  24. )
  25. from torch.ao.quantization.fx.graph_module import (
  26. ObservedGraphModule,
  27. )