pytorch
4483c48e - ns for fx: support linear_relu for weight matching (#52395)

Commit
3 years ago
ns for fx: support linear_relu for weight matching (#52395) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/52395 Simple change to add logic to get the weight of a quantized `linear_relu` node. More flavors of conv and linear will be added in future PRs. Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs.test_compare_weights_fun ``` Imported from OSS Reviewed By: hx89 Differential Revision: D26497992 fbshipit-source-id: e6d88e92eedd6cdbf9116cbcfc8f6164f8499246
Author
Parents
Loading