Remove casting of a list on node.args & add torch.ops.aten.full.default

back to is_share_obs_or_fq_op

Author: SaoirseARM

backends/arm/quantizer/arm_quantizer_utils.py

@@ -147,6 +147,7 @@ def is_share_obs_or_fq_op(op: Callable) -> bool:
         # TODO: remove?
         torch.ops.aten.adaptive_avg_pool2d.default,
         torch.ops.aten.avg_pool2d.default,
+        torch.ops.aten.full.default,
         torch.ops.aten.flatten.using_ints,
         torch.ops.aten.dropout.default,
         operator.getitem,

backends/arm/quantizer/quantization_annotation/generic_annotator.py

@@ -4,7 +4,7 @@
 # LICENSE file in the root directory of this source tree.
 
 # pyre-unsafe
-from typing import Callable, cast, List, Optional
+from typing import Callable, List, Optional
 
 import torch
 import torch.fx
@@ -70,7 +70,7 @@ def _annotate_generic(
         if arm_quantizer_utils.is_annotated(node):
             continue
 
-        input_acts = cast(list[torch.fx.Node], node.args[0])
+        input_acts = node.args[0]
 
         # Check to see if there are multiple inputs.
         # this allows for stack/cat ops to be annotated