pytorch/test/quantization
vspenubarthi 0656e9e595 [ao] Adding model report detector base class and implemented detectors
Summary: the goal is to add a base class to the model report detectors
so that they can contain a lot more specific information compared to the
primary model report class related to the observers and where they are
inserted etc.

Since this is just a base class, the testing will be with the
implemenations of the classes that derive from the base class

The two current detector methods were turned into Detector classes and
the tests were modified to reflect this, but the same functionality was
tested.

As a result, _detector.py was changed to detector.py

Test Plan: python test/test_quantization.py TestFxModelReportDetector
python test/test_quantization.py TestFxModelReportDetectDynamicStatic

Reviewers:

Subscribers:

Tasks:

Tags:

Pull Request resolved: https://github.com/pytorch/pytorch/pull/79671

Approved by: https://github.com/andrewor14
2022-06-21 00:21:50 +00:00
..
ao_migration fx quant: refactor qconfig setting out of find_matches 2022-06-17 18:52:00 +00:00
bc [quant][fx][bc-breaking] Add required example_args argument to prepare_fx and prepare_qat_fx (#249) (#77608) 2022-05-21 21:03:48 +00:00
core [quant] Add param calcs and tests for APoT observers 2022-06-15 23:24:48 +00:00
dbr [quant][fx][bc-breaking] Add required example_args argument to prepare_fx and prepare_qat_fx (#249) (#77608) 2022-05-21 21:03:48 +00:00
eager Fix ONNX ATen fallback for non-caffe2 engines 2022-04-14 23:18:45 +00:00
fx [ao] Adding model report detector base class and implemented detectors 2022-06-21 00:21:50 +00:00
jit extend replaceConvolutionWithAtenConv to handle conv_transpose3d (#76888) 2022-05-13 16:40:12 +00:00
serialized
__init__.py