Skip to content

Commit bb4003c

Browse files
committed
Arm backend: Remove aten.zeroes from annotator
Removing torch.ops.aten.zeros from the quantization_annotator. Signed-off-by: [email protected] Change-Id: Id18089846a0447a480e547578ab062ad42dd1065
1 parent 04c69cf commit bb4003c

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

backends/arm/quantizer/quantization_annotator.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -697,7 +697,6 @@ def any_or_hardtanh_min_zero(n: Node):
697697
elif node.target in [
698698
torch.ops.aten.full.default,
699699
torch.ops.aten.full,
700-
torch.ops.aten.zeros.default,
701700
torch.ops.aten.ones.default,
702701
torch.ops.aten.fill_.Scalar,
703702
torch.ops.aten.scalar_tensor.default,
@@ -798,7 +797,6 @@ def annotate_graph( # type: ignore[return]
798797
torch.ops.aten.full,
799798
torch.ops.aten.fill_.Scalar,
800799
torch.ops.aten.scalar_tensor.default,
801-
torch.ops.aten.zeros.default,
802800
torch.ops.aten.ones.default,
803801
]:
804802
node.kwargs = {}

0 commit comments

Comments
 (0)