OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
ffc0c46092
pytorch/torch/ao/quantization
History
Kimish Patel ffc0c46092 [Quantization] Add metadata porting for nodes added by quantization (#107107) 
Summary:
This diff adds adding metadata to q-dq nodes by inferring the
quatization intent from node annotations. Annotations on the node are
way for user to specify how a node or subgraph is supposed to be
quantized. We continue to use that information to copy metadata on Q/DQ
node from appropriate nodes.

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

Differential Revision: [D48488416](https://our.internmc.facebook.com/intern/diff/D48488416)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/107107
Approved by: https://github.com/jerryzh168
ghstack dependencies: #107105, #107106, #107899, #107900
2023-09-02 06:38:14 +00:00
..
backend_config [Quant][PT2E] Remove x86 inductor pt2e backend config (#105039) 2023-07-19 23:18:29 +00:00
experimental [BE] Enable ruff's UP rules and autoformat ao/ (#105430) 2023-07-19 13:44:37 +00:00
fx [quant][pt2e] Add support for channel in DerivedQuantizationSpec (#107833) 2023-08-24 07:45:13 +00:00
pt2e [Quantization] Add metadata porting for nodes added by quantization (#107107) 2023-09-02 06:38:14 +00:00
quantizer [Quant] Add DQ duplication pass (#107900) 2023-09-02 06:20:03 +00:00
__init__.py [Quant] Add DQ duplication pass (#107900) 2023-09-02 06:20:03 +00:00
_correct_bias.py [BE] [2/3] Rewrite super() calls in functorch and torch (#94588) 2023-02-10 21:16:33 +00:00
_equalize.py Fix typos under torch/ao directory (#97679) 2023-04-10 22:25:15 +00:00
_learnable_fake_quantize.py [BE] Enable ruff's UP rules and autoformat ao/ (#105430) 2023-07-19 13:44:37 +00:00
fake_quantize.py [BE] Enable ruff's UP rules and autoformat ao/ (#105430) 2023-07-19 13:44:37 +00:00
fuse_modules.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
fuser_method_mappings.py [BE] Enable ruff's UP rules and autoformat ao/ (#105430) 2023-07-19 13:44:37 +00:00
observer.py [PyPer][BE] Fix test_scripted_module in StatCollector (#108232) 2023-09-01 04:23:57 +00:00
pattern.md
qconfig_mapping.py [ao] fixing quantized prelu workflow (#103455) 2023-06-23 16:45:40 +00:00
qconfig.py [BE] Enable ruff's UP rules and autoformat ao/ (#105430) 2023-07-19 13:44:37 +00:00
quant_type.py [BE] Enable ruff's UP rules and autoformat ao/ (#105430) 2023-07-19 13:44:37 +00:00
quantization_mappings.py [BE] Enable ruff's UP rules and autoformat ao/ (#105430) 2023-07-19 13:44:37 +00:00
quantize_fx.py [quant][fx] Fix docs for prepare_fx/prepare_qat_fx (#105979) 2023-07-26 09:56:18 +00:00
quantize_jit.py Fix typos under torch/ao directory (#97679) 2023-04-10 22:25:15 +00:00
quantize_pt2e.py [Quantization] Add metadata porting for nodes added by quantization (#107107) 2023-09-02 06:38:14 +00:00
quantize.py [BE] Enable ruff's UP rules and autoformat ao/ (#105430) 2023-07-19 13:44:37 +00:00
stubs.py [codemod] Replace hasattr with getattr in caffe2/torch/ao/quantization/stubs.py (#100597) 2023-05-04 16:36:23 +00:00
utils.py [BE]: Update ruff to 0.285 (#107519) 2023-08-22 23:16:38 +00:00