commit f9ca48ddb58196fa2003d6253c38c84567893e34
author maxren <maxren@meta.com> Thu Mar 30 09:11:46 2023 -0700
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Fri Mar 31 04:58:24 2023 +0000
tree 9d3e0edc87af35f6439b2fd613d2271057b44a26
parent ae5b044ccb826793e91272e14f3feecf57ac085f

[Executorch][XNNPACK] Quantized hardtanh (#97387)

Lower Quantized Hardtanh to XNNPACK

Also add symmetric quantization support for hardtanh in executorch backend config

Differential Revision: [D43901222](https://our.internmc.facebook.com/intern/diff/D43901222/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/97387
Approved by: https://github.com/salilsdesai

torch/ao/quantization/backend_config/executorch.py

diff --git a/torch/ao/quantization/backend_config/executorch.py b/torch/ao/quantization/backend_config/executorch.py
index cd4df7f..1b9e31e 100644
--- a/torch/ao/quantization/backend_config/executorch.py
+++ b/torch/ao/quantization/backend_config/executorch.py
@@ -218,9 +218,11 @@
     ]
     share_qparams_ops = [
         F.adaptive_avg_pool2d,
+        F.hardtanh,
         F.relu,
         F.relu6,
         torch.nn.AdaptiveAvgPool2d,
+        torch.nn.Hardtanh,
         torch.squeeze,
         "permute",
         "reshape",