Lingyi Liu
fddcd72a31
Add the more fusion (conv3d and batchnorm)support in pytorch quantization flow ( #33540 )
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/33540
Differential Revision: D19994498
Pulled By: lly-zero-one
fbshipit-source-id: e5e13eab6924bd2ce1b57b16b672844b8b9638f5
2020-03-23 20:36:03 -07:00
Lingyi Liu
68758b2fa0
Add the quantized batch_norm3d and also batch_norm3d fused with relu operators ( #34702 )
...
Summary:
as title, for bringing up the quantized video model. Will add the batch_norm_relu test in another PR.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/34702
Differential Revision: D20436092
Pulled By: lly-zero-one
fbshipit-source-id: 116bd306f7880bfd763d8575654fbd6c92818338
2020-03-13 20:30:28 -07:00
Lingyi Liu
f70945b1c3
fix the quantized batchnorm2d ( #34579 )
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/34579
Differential Revision: D20382783
Pulled By: lly-zero-one
fbshipit-source-id: dadfc4974cb4c808f1eedf8cc4ec52ec8d3ea1b0
2020-03-12 00:48:40 -07:00
Supriya Rao
2e88d3d703
[quant] Add Quantized BatchNorm2d module ( #33109 )
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/33109
Test Plan:
python test/test_quantized_nn_mods.py ModuleAPITest.test_batch_norm
Imported from OSS
Differential Revision: D19861926
fbshipit-source-id: 67315e49b4b3577b965d422ca707d927d977feeb
2020-02-13 12:15:43 -08:00
