[quant][fx2trt] Add lowering support for reference linear/conv modules (#64368)
authorJerry Zhang <jerryzh@fb.com>
Sat, 11 Sep 2021 05:24:10 +0000 (22:24 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Sat, 11 Sep 2021 05:25:27 +0000 (22:25 -0700)
commitd4a86c1f3b84aabb1a903b856190556a2f087b81
tree9b9defcb7bbb9406711e1f5f51d225ceef43e17a
parent4481c87ac408560cb03aa3c81d0dcc1a77acdd4c
[quant][fx2trt] Add lowering support for reference linear/conv modules (#64368)

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

Test Plan:
python torch/fx/experimental/fx2trt/example/quantized_resnet_test.py

Imported from OSS

Reviewed By: 842974287

Differential Revision: D30708738

fbshipit-source-id: 88142b7ce43ed96093597112dab03a2d277de993
torch/fx/experimental/fx2trt/converters/acc_ops_converters.py
torch/fx/experimental/fx2trt/example/quantized_resnet_test.py
torch/fx/experimental/fx2trt/fx2trt.py
torch/nn/quantized/_reference/modules/conv.py
torch/nn/quantized/_reference/modules/linear.py
torch/quantization/fx/quantization_patterns.py