Vasiliy Kuznetsov 07c6a12008 ns for fx: fix typing issue in weight extraction (#62041) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/62041 Before this PR, weights of conv and linear modules were extracted as lists, in order to match the signature of LSTM weights. After this PR, weight extraction preserves the type of the weights, so extracted weights of conv and linear have a different type from LSTM weights. The comparison util functions are updated to handle the LSTM weight type of `List[tensor]`. Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs python test/test_quantization.py TestFXNumericSuiteCoreAPIsModels ``` Imported from OSS Reviewed By: jerryzh168 Differential Revision: D29853626 fbshipit-source-id: 93da5b9b0b174679c61528d02b6b902cb064444e		2021-07-23 09:31:33 -07:00
..
_internal	ns for fx: fix typing issue in weight extraction (#62041)	2021-07-23 09:31:33 -07:00
__init__.py	Un-ignore F403 in .flake8 (#55838)	2021-04-13 09:24:07 -07:00
_asserts.py	Add neg bit (#56058)	2021-07-13 13:50:42 -07:00
_check_kernel_launches.py	Paren-matching kernel launch check without external deps (#60778)	2021-06-28 10:18:04 -07:00
_core.py	Revert D29361872: [pytorch][PR] det_backward: more robust and with complex support	2021-07-15 15:26:00 -07:00