diff --git a/backends/arm/quantizer/quantization_annotator.py b/backends/arm/quantizer/quantization_annotator.py index 4a699e9f3eb..5b5c6a6f5d6 100644 --- a/backends/arm/quantizer/quantization_annotator.py +++ b/backends/arm/quantizer/quantization_annotator.py @@ -603,6 +603,10 @@ def _get_fixed_qparams_qspec( torch.ops.aten.split_copy.Tensor, torch.ops.aten.transpose.int, torch.ops.aten.transpose_copy.int, + torch.ops.aten.moveaxis.int, + torch.ops.aten.moveaxis.intlist, + torch.ops.aten.movedim.int, + torch.ops.aten.movedim.intlist, torch.ops.aten.t_copy.default, torch.ops.aten.tile.default, torch.ops.aten.flip.default, @@ -628,25 +632,6 @@ def _get_fixed_qparams_qspec( torch.ops.aten.detach_copy.default, } -# Dimname has been removed from upstream PyTorch, but there may be a window -# where developers in this backend are using a mainline build of this backend -# with an older version of PyTorch. -# TODO: remove this once the build has time to be propagated and majority of -# dev expected to be unimpacted -_transpose_dimname = getattr(torch.ops.aten.transpose, "Dimname", None) -if _transpose_dimname is not None: - _one_to_one_shared_input_qspec.add(_transpose_dimname) - -for _op in ( - getattr(torch.ops.aten.moveaxis, "int", None), - getattr(torch.ops.aten.moveaxis, "intlist", None), - getattr(torch.ops.aten.movedim, "int", None), - getattr(torch.ops.aten.movedim, "intlist", None), -): - if _op is not None: - _one_to_one_shared_input_qspec.add(_op) - - _one_to_one_shared_input_or_input_act_qspec: set[OpOverload] = { torch.ops.aten.alias.default, torch.ops.aten.clone.default,