Skip to content

Commit 71af9b9

Browse files
committed
add slice op in _is_share_obs_or_fq_op function
1 parent 9591978 commit 71af9b9

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

backends/xnnpack/quantizer/xnnpack_quantizer_utils.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1060,6 +1060,9 @@ def _is_share_obs_or_fq_op(op: Callable) -> bool:
10601060
torch.ops.aten.view.default,
10611061
torch.ops.aten.slice_copy.Tensor,
10621062
torch.ops.aten.flatten.using_ints,
1063+
torch.ops.aten.slice.Tensor,
1064+
torch.ops.aten.select.int,
1065+
torch.ops.aten.index_select.default,
10631066
]
10641067

10651068

0 commit comments

Comments
 (0)