Skip to content

Commit 68eb62f

Browse files
authored
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]>
1 parent 75d4abc commit 68eb62f

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

backends/arm/quantizer/quantization_annotator.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -183,6 +183,7 @@ def _match_pattern(
183183
torch.ops.aten.hardtanh.default,
184184
torch.ops.aten.hardtanh_.default,
185185
torch.ops.aten.relu.default,
186+
torch.ops.aten.relu_.default,
186187
torch.ops.aten.mean.default,
187188
torch.ops.aten.mean.dim,
188189
torch.ops.aten.permute.default,

0 commit comments

Comments
 (0)