pytorch
f6192b75 - [Quant] Support lowering of channel shuffle in FX (#83731)

Commit
2 years ago
[Quant] Support lowering of channel shuffle in FX (#83731) ## Description Support lowering of channel shuffle in FX by adding its module and functional op to `is_copy_node` list in `torch/ao/quantization/fx/_lower_to_native_backend.py` ## Validation UTs added to test - correctness of quantized `ChannelShuffle` module. - FX lowering of `ChannelShuffle` module and functional `channel_shuffle`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/83731 Approved by: https://github.com/jerryzh168
Author
Committer
Parents
Loading