diff --git a/backends/arm/operator_support/tosa_supported_operators.py b/backends/arm/operator_support/tosa_supported_operators.py index 9fcff7bdc5f..437e13edfb2 100644 --- a/backends/arm/operator_support/tosa_supported_operators.py +++ b/backends/arm/operator_support/tosa_supported_operators.py @@ -327,6 +327,7 @@ class CheckProperQuantization(OperatorSupportBase): exir_ops.edge.aten.upsample_bilinear2d.vec, exir_ops.edge.aten.upsample_nearest2d.vec, torch.ops.aten.scalar_tensor.default, + exir_ops.edge.aten.mean.dim, *TableOps.included_ops(), )