xref: /aosp_15_r20/external/pytorch/torch/quantization/observer.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1*da0073e9SAndroid Build Coastguard Worker# flake8: noqa: F401
2*da0073e9SAndroid Build Coastguard Workerr"""
3*da0073e9SAndroid Build Coastguard WorkerThis file is in the process of migration to `torch/ao/quantization`, and
4*da0073e9SAndroid Build Coastguard Workeris kept here for compatibility while the migration process is ongoing.
5*da0073e9SAndroid Build Coastguard WorkerIf you are adding a new entry/functionality, please, add it to the
6*da0073e9SAndroid Build Coastguard Worker`torch/ao/quantization/observer.py`, while adding an import statement
7*da0073e9SAndroid Build Coastguard Workerhere.
8*da0073e9SAndroid Build Coastguard Worker"""
9*da0073e9SAndroid Build Coastguard Workerfrom torch.ao.quantization.observer import (
10*da0073e9SAndroid Build Coastguard Worker    _is_activation_post_process,
11*da0073e9SAndroid Build Coastguard Worker    _is_per_channel_script_obs_instance,
12*da0073e9SAndroid Build Coastguard Worker    _ObserverBase,
13*da0073e9SAndroid Build Coastguard Worker    _PartialWrapper,
14*da0073e9SAndroid Build Coastguard Worker    _with_args,
15*da0073e9SAndroid Build Coastguard Worker    _with_callable_args,
16*da0073e9SAndroid Build Coastguard Worker    ABC,
17*da0073e9SAndroid Build Coastguard Worker    default_debug_observer,
18*da0073e9SAndroid Build Coastguard Worker    default_dynamic_quant_observer,
19*da0073e9SAndroid Build Coastguard Worker    default_float_qparams_observer,
20*da0073e9SAndroid Build Coastguard Worker    default_histogram_observer,
21*da0073e9SAndroid Build Coastguard Worker    default_observer,
22*da0073e9SAndroid Build Coastguard Worker    default_per_channel_weight_observer,
23*da0073e9SAndroid Build Coastguard Worker    default_placeholder_observer,
24*da0073e9SAndroid Build Coastguard Worker    default_weight_observer,
25*da0073e9SAndroid Build Coastguard Worker    get_observer_state_dict,
26*da0073e9SAndroid Build Coastguard Worker    HistogramObserver,
27*da0073e9SAndroid Build Coastguard Worker    load_observer_state_dict,
28*da0073e9SAndroid Build Coastguard Worker    MinMaxObserver,
29*da0073e9SAndroid Build Coastguard Worker    MovingAverageMinMaxObserver,
30*da0073e9SAndroid Build Coastguard Worker    MovingAveragePerChannelMinMaxObserver,
31*da0073e9SAndroid Build Coastguard Worker    NoopObserver,
32*da0073e9SAndroid Build Coastguard Worker    ObserverBase,
33*da0073e9SAndroid Build Coastguard Worker    PerChannelMinMaxObserver,
34*da0073e9SAndroid Build Coastguard Worker    PlaceholderObserver,
35*da0073e9SAndroid Build Coastguard Worker    RecordingObserver,
36*da0073e9SAndroid Build Coastguard Worker)
37