From ee470486e0d14bc8071de6f45280afb2d2b4cb0a Mon Sep 17 00:00:00 2001 From: Aleksei Kashapov Date: Mon, 29 Jul 2024 11:32:52 +0200 Subject: [PATCH] add slice metatype --- nncf/openvino/graph/metatypes/groups.py | 1 + nncf/openvino/graph/metatypes/openvino_metatypes.py | 6 ++++++ 2 files changed, 7 insertions(+) diff --git a/nncf/openvino/graph/metatypes/groups.py b/nncf/openvino/graph/metatypes/groups.py index c19d2bae83c..806b3d93367 100644 --- a/nncf/openvino/graph/metatypes/groups.py +++ b/nncf/openvino/graph/metatypes/groups.py @@ -22,6 +22,7 @@ ov_metatypes.OVTransposeMetatype, ov_metatypes.OVTileMetatype, ov_metatypes.OVStridedSliceMetatype, + ov_metatypes.OVSliceMetatype, ov_metatypes.OVShuffleChannelsMetatype, ov_metatypes.OVBroadcastMetatype, ov_metatypes.OVPadMetatype, diff --git a/nncf/openvino/graph/metatypes/openvino_metatypes.py b/nncf/openvino/graph/metatypes/openvino_metatypes.py index 424ef81b914..430bc14b49c 100644 --- a/nncf/openvino/graph/metatypes/openvino_metatypes.py +++ b/nncf/openvino/graph/metatypes/openvino_metatypes.py @@ -565,6 +565,12 @@ class OVStridedSliceMetatype(OVOpMetatype): hw_config_names = [HWConfigOpName.STRIDEDSLICE] +@OV_OPERATOR_METATYPES.register() +class OVSliceMetatype(OVOpMetatype): + name = "SliceOp" + op_names = ["Slice"] + + @OV_OPERATOR_METATYPES.register() class OVExpMetatype(OVOpMetatype): name = "ExpOp"