observer.py 1.1 KB

123456789101112131415161718192021222324252627282930313233343536
  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/observer.py`, while adding an import statement
  7. here.
  8. """
  9. from torch.ao.quantization.observer import (
  10. _PartialWrapper,
  11. _with_args,
  12. _with_callable_args,
  13. ABC,
  14. ObserverBase,
  15. _ObserverBase,
  16. MinMaxObserver,
  17. MovingAverageMinMaxObserver,
  18. PerChannelMinMaxObserver,
  19. MovingAveragePerChannelMinMaxObserver,
  20. HistogramObserver,
  21. PlaceholderObserver,
  22. RecordingObserver,
  23. NoopObserver,
  24. _is_activation_post_process,
  25. _is_per_channel_script_obs_instance,
  26. get_observer_state_dict,
  27. load_observer_state_dict,
  28. default_observer,
  29. default_placeholder_observer,
  30. default_debug_observer,
  31. default_weight_observer,
  32. default_histogram_observer,
  33. default_per_channel_weight_observer,
  34. default_dynamic_quant_observer,
  35. default_float_qparams_observer,
  36. )