pytorch
c3d4053b - [quant][graphmode] Support quantized::batch_norm2d_relu fusion for tracing (#39645)

Commit
4 years ago
[quant][graphmode] Support quantized::batch_norm2d_relu fusion for tracing (#39645) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/39645 This PR added quantization support for handling BatchNorm2d and ReLU(or F.relu) in both scripting and tracing Test Plan: python test/test_quantization.py TestQuantizeScriptPTSQOps.test_qbatchnorm_relu Imported from OSS Differential Revision: D21942111 fbshipit-source-id: 680e16076a37b96d2485d5cbc39ce9a045c319c3
Author
Parents
Loading