pytorch
cb231915 - [Vulkan] rewrite quantized add, mul, conv2d and conv2d_relu ops (#97468)

Commit
1 year ago
[Vulkan] rewrite quantized add, mul, conv2d and conv2d_relu ops (#97468) Summary: This diffs registers the vulkan quantized binary ops (add/sub/mul/div), and adds graph rewrites for quantized add, mul, conv2d and conv2d_relu. The rewrites for conv2d and conv2d_relu make use of the convert_qconv2d_context introduced in D41595032 Test Plan: export quantized mcs model to vulkan Reviewed By: SS-JIA Differential Revision: D44189363 Pull Request resolved: https://github.com/pytorch/pytorch/pull/97468 Approved by: https://github.com/SS-JIA
Committer
Parents
Loading