pytorch/_quantize_fx.py at dae2973faeb2f4065297dbfa27def8b1e2746569 - pytorch - Carlos Sousa's Git

OSSForks/pytorch

mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00

Jerry Zhang dae2973fae [quant][graphmode][fx] Add graph mode quantization on fx (#43175 )

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/43175

This PR added graph mode quantization on fx: https://github.com/pytorch/pytorch/pull/42741
Currently it matches eager mode quantization for torchvision with static/dynamic/qat
ddp/synbn test is still wip

Test Plan:
python test/test_quantization.py TestQuantizeFx

Imported from OSS

Reviewed By: vkuzo

Differential Revision: D23178602

fbshipit-source-id: 8e7e0322846fbda2cfa79ad188abd7235326f879

2020-08-20 14:50:09 -07:00

3 lines

75 B

Python

Raw Blame History

	`from .fx import Quantizer # noqa: F401`
	`from .fx import fuse # noqa: F401`