diff --git a/nncf/common/hardware/configs/cpu.json b/nncf/common/hardware/configs/cpu.json index 0cd5290ae6d..ae6a6645044 100644 --- a/nncf/common/hardware/configs/cpu.json +++ b/nncf/common/hardware/configs/cpu.json @@ -279,6 +279,7 @@ {"type": "Transpose"}, {"type": "Tile"}, {"type": "StridedSlice"}, + {"type": "Slice"}, {"type": "ShuffleChannels"}, {"type": "Broadcast"}, {"type": "Pad"}, diff --git a/nncf/common/hardware/configs/gpu.json b/nncf/common/hardware/configs/gpu.json index 9e6d0e3378b..39cb1cc6573 100644 --- a/nncf/common/hardware/configs/gpu.json +++ b/nncf/common/hardware/configs/gpu.json @@ -247,6 +247,7 @@ {"type": "Crop"}, {"type": "Transpose"}, {"type": "Tile"}, - {"type": "StridedSlice"} + {"type": "StridedSlice"}, + {"type": "Slice"} ] } diff --git a/nncf/common/hardware/configs/npu.json b/nncf/common/hardware/configs/npu.json index 7999bd874e9..15af02a9f60 100644 --- a/nncf/common/hardware/configs/npu.json +++ b/nncf/common/hardware/configs/npu.json @@ -372,6 +372,7 @@ {"type": "Transpose"}, {"type": "Tile"}, {"type": "StridedSlice"}, + {"type": "Slice"}, {"type": "ShuffleChannels"}, {"type": "Broadcast"}, {"type": "Pad"}, diff --git a/nncf/common/hardware/opset.py b/nncf/common/hardware/opset.py index c01e950b417..91fafe0b09c 100644 --- a/nncf/common/hardware/opset.py +++ b/nncf/common/hardware/opset.py @@ -54,6 +54,7 @@ class HWConfigOpName: EMBEDDINGBAG = "EmbeddingBag" PAD = "Pad" STRIDEDSLICE = "StridedSlice" + SLICE = "Slice" GELU = "Gelu" LSTMSEQUENCE = "LSTMSequence" GRUSEQUENCE = "GRUSequence" 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..87a2677c0d7 100644 --- a/nncf/openvino/graph/metatypes/openvino_metatypes.py +++ b/nncf/openvino/graph/metatypes/openvino_metatypes.py @@ -565,6 +565,13 @@ class OVStridedSliceMetatype(OVOpMetatype): hw_config_names = [HWConfigOpName.STRIDEDSLICE] +@OV_OPERATOR_METATYPES.register() +class OVSliceMetatype(OVOpMetatype): + name = "SliceOp" + op_names = ["Slice"] + hw_config_names = [HWConfigOpName.SLICE] + + @OV_OPERATOR_METATYPES.register() class OVExpMetatype(OVOpMetatype): name = "ExpOp"