graph_module.py 572 B

1234567891011121314151617
  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. appropriate files under `torch/ao/quantization/fx/`, while adding an import statement
  7. here.
  8. """
  9. from torch.ao.quantization.fx.graph_module import (
  10. GraphModule,
  11. FusedGraphModule,
  12. ObservedGraphModule,
  13. _is_observed_module,
  14. ObservedStandaloneGraphModule,
  15. _is_observed_standalone_module,
  16. QuantizedGraphModule
  17. )