From ff24a6cefc270d9c9e589af5a32977a7ed971910 Mon Sep 17 00:00:00 2001 From: Val Tarasyuk Date: Fri, 22 Nov 2024 10:49:15 -0800 Subject: [PATCH] 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 --- backends/cadence/aot/remove_ops.py | 1 + 1 file changed, 1 insertion(+) diff --git a/backends/cadence/aot/remove_ops.py b/backends/cadence/aot/remove_ops.py index d2251bd9c08..038a2192079 100644 --- a/backends/cadence/aot/remove_ops.py +++ b/backends/cadence/aot/remove_ops.py @@ -561,6 +561,7 @@ class Subgraph: exir_ops.edge.aten.mul.Tensor, exir_ops.edge.aten.mean.dim, exir_ops.edge.aten.cat.default, + exir_ops.edge.aten.hardtanh.default, exir_ops.edge.quantized_decomposed.quantize_per_tensor.default, exir_ops.edge.quantized_decomposed.dequantize_per_tensor.default, }