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. _is_activation_post_process,
  11. _is_per_channel_script_obs_instance,
  12. _ObserverBase,
  13. _PartialWrapper,
  14. _with_args,
  15. _with_callable_args,
  16. ABC,
  17. default_debug_observer,
  18. default_dynamic_quant_observer,
  19. default_float_qparams_observer,
  20. default_histogram_observer,
  21. default_observer,
  22. default_per_channel_weight_observer,
  23. default_placeholder_observer,
  24. default_weight_observer,
  25. get_observer_state_dict,
  26. HistogramObserver,
  27. load_observer_state_dict,
  28. MinMaxObserver,
  29. MovingAverageMinMaxObserver,
  30. MovingAveragePerChannelMinMaxObserver,
  31. NoopObserver,
  32. ObserverBase,
  33. PerChannelMinMaxObserver,
  34. PlaceholderObserver,
  35. RecordingObserver,
  36. )