pytorch
2bc56bec - [quant][pt2] Handle literal conv args in convert QAT (#103731)

Commit
1 year ago
[quant][pt2] Handle literal conv args in convert QAT (#103731) Summary: Similar to the prepare case, we need to manually copy over literal conv args such as padding and stride to the new, replaced conv nodes, since these args are not captured by the subgraph rewriter. Test Plan: python test/test_quantization.py TestQuantizePT2E.test_qat_conv_bn_fusion_literal_args Reviewed By: jerryzh168 Differential Revision: D46383130 Pull Request resolved: https://github.com/pytorch/pytorch/pull/103731 Approved by: https://github.com/jerryzh168
Author
Committer
Parents
Loading