pytorch
9a0b7b47 - [quant] Fix implementation for `output_quantized_idxs` in convert (#74140) (#74229)

Commit
3 years ago
[quant] Fix implementation for `output_quantized_idxs` in convert (#74140) (#74229) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/74229 Previously we did not successfully remove the dequantize node for `dict`, this PR fixes that, tested with meta-only tests right now but we should follow up with oss tests (with dict output) since we called dead code elimination pass, some of the inplace operators are removed in the TestQuantizeFx.test_fixed_qparams_ops, in this PR we also just removed the calls to the inplace ops, and changed the expected results in the test case, in the future PR we can remove the support for inplace operators, since it is not really supported in fx, and it's OK for us to skip them as well Test Plan: Imported from OSS Reviewed By: vkuzo Differential Revision: D34888140 fbshipit-source-id: 48cea842b49e52baa8eee3ce0f4bfb4a3625ab2a (cherry picked from commit ef790315ebcf954930deb6b9d1c384992c1f1ec8)
Author
Committer
Parents
Loading