Jon Morton 123be0e5b7 [fusion] Add ConvTranspose+BN fusion support (#70022) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/70022 Add support for fusing ConvTranpose{1,2,3}d with BatchNorm{1,2,3}d. This re-uses the existing fusion logic but adds a "transpose" flag to the fusing function which when enabled will use the appropriate reshape for ConTranspose's transposed weights. Test Plan: `buck test mode/dev //caffe2/test:quantization -- -r quantization.eager.test_fusion.TestFusion` Reviewed By: jerryzh168 Differential Revision: D33074405 fbshipit-source-id: 5e9eff1a06d8f98d117e7d18e80da8e842e973b7		2021-12-20 18:42:48 -08:00
..
__init__.py
_stateless.py
clip_grad.py	[Docs] Fix the syntax of documentation (#69958)	2021-12-16 10:38:39 -08:00
convert_parameters.py
fusion.py	[fusion] Add ConvTranspose+BN fusion support (#70022)	2021-12-20 18:42:48 -08:00
init.py
memory_format.py
parametrizations.py
parametrize.py
prune.py
rnn.py
rnn.pyi
spectral_norm.py
weight_norm.py