pytorch
15271d35 - [quant][pt2e] Support convtranspose + bn fusion (#97933)

Commit
2 years ago
[quant][pt2e] Support convtranspose + bn fusion (#97933) Summary: This PR extends `_fuse_conv_bn_` function to support fusing convtranspose and bn Test Plan: python test/test_quantization.py TestQuantizePT2E.test_transposed_conv_bn_fusion Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/97933 Approved by: https://github.com/vkuzo
Author
Committer
Parents
Loading