[quant][graphmode][fix] Fold conv bn (#40865) (#40970)
* [quant][graphmode][fix] Fold conv bn (#40865)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40865
1. applied filter for the module types
2. removed the assumption that the conv bn are immediate child of parent module
Test Plan:
python test/test_quantization.py TestQuantizeJitPasses
Imported from OSS
Differential Revision: D22338074
fbshipit-source-id: 64739a5e56c0a74249a1dbc2c8454b88ec32aa9e
* [quant][graphmode][fix] Print the node in error message (#40889)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/40889
Test Plan: Imported from OSS
Differential Revision: D22348266
fbshipit-source-id: eed2ece5c94fcfaf187d6770bed4a7109f0c0b4a
