Skip to content

Commit ff24a6c

Browse files
Vysaratfacebook-github-bot
authored andcommitted
Add HardTanh to RemovePermutesAroundElementwiseOps (#6992)
Summary: This diff adds `hardtanh` to the list of elementwise ops that we consider for permute elimination. - Fixing logging issue in summary dump that results in some delegate size metrics being reported incorrectly. Reviewed By: zonglinpeng Differential Revision: D66187338
1 parent fc42a4e commit ff24a6c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

backends/cadence/aot/remove_ops.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -561,6 +561,7 @@ class Subgraph:
561561
exir_ops.edge.aten.mul.Tensor,
562562
exir_ops.edge.aten.mean.dim,
563563
exir_ops.edge.aten.cat.default,
564+
exir_ops.edge.aten.hardtanh.default,
564565
exir_ops.edge.quantized_decomposed.quantize_per_tensor.default,
565566
exir_ops.edge.quantized_decomposed.dequantize_per_tensor.default,
566567
}

0 commit comments

Comments
 (0)