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
72c943a2ac
pytorch/torch/quantization
History
Vasiliy Kuznetsov 72c943a2ac ns for fx: fix bug for user function in weight extraction (#62333) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/62333

We incorrectly ignored any custom relationships the user specified
in the `extract_weights` API.  Fixing this and adding a test case.

Test Plan:
```
python test/test_quantization.py TestFXNumericSuiteCoreAPIs.test_user_defined_function
```

Imported from OSS

Reviewed By: hx89

Differential Revision: D29963502

fbshipit-source-id: 33ce3d4df1acb6298b6c7dcb6674015c8d14bdf4
2021-07-28 16:05:51 -07:00
..
fx [quant][graphmode][fx] Add reference pattern support for BatchNorm (#62215) 2021-07-28 10:14:16 -07:00
ns ns for fx: expose hook to define custom weight extraction functions (#62047) 2021-07-23 09:31:37 -07:00
__init__.py Un-ignore F403 in .flake8 (#55838) 2021-04-13 09:24:07 -07:00
_correct_bias.py
_equalize.py [quant] Eager mode equalization support for ConvReLU and LinearReLU (#58792) 2021-05-24 17:25:13 -07:00
_learnable_fake_quantize.py [docs] Fix backticks in docs (#60474) 2021-06-24 06:27:41 -07:00
_numeric_suite_fx.py ns for fx: fix bug for user function in weight extraction (#62333) 2021-07-28 16:05:51 -07:00
_numeric_suite.py Add lint for unqualified type: ignore (#56290) 2021-04-21 08:07:23 -07:00
fake_quantize.py [quant] Create FusedMovingAvgObsFakeQuantize for QAT (#61691) 2021-07-21 10:13:04 -07:00
fuse_modules.py Add lint for unqualified noqa (#56272) 2021-04-19 13:16:18 -07:00
fuser_method_mappings.py fix docstring for fusing functions (#58638) 2021-05-24 18:27:22 -07:00
observer.py Support for reference convert_fx working on gpu 2021-07-23 10:30:38 -07:00
qconfig.py Support for reference convert_fx working on gpu 2021-07-23 10:30:38 -07:00
quant_type.py
quantization_mappings.py [quant][graphmode][fx] Produce reference linear module in convert (#60152) 2021-06-20 20:08:12 -07:00
quantize_fx.py [quant] Input Weight Equalization - prepare modifications (#59747) 2021-06-16 22:32:28 -07:00
quantize_jit.py Enable the quantization on XPU devices (#54857) 2021-05-20 17:02:13 -07:00
quantize.py Support for reference convert_fx working on gpu 2021-07-23 10:30:38 -07:00
stubs.py
utils.py [quant] Equalization Observer modifications (#59953) 2021-06-16 22:32:30 -07:00