Arm backend: Add aten.relu_.default to quantization_annotator (#8567)

* The inplace version of relu (aten.relu_.default) was missing Signed-off-by: Tom Allsop <[email protected]>
pytorch · Feb 20, 2025 · 68eb62f · 68eb62f
1 parent 75d4abc
commit 68eb62f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/backends/arm/quantizer/quantization_annotator.py b/backends/arm/quantizer/quantization_annotator.py
@@ -183,6 +183,7 @@ def _match_pattern(
     torch.ops.aten.hardtanh.default,
     torch.ops.aten.hardtanh_.default,
     torch.ops.aten.relu.default,
+    torch.ops.aten.relu_.default,
     torch.ops.aten.mean.default,
     torch.ops.aten.mean.dim,
     torch.ops.aten.permute.default,