diff --git a/nncf/openvino/graph/metatypes/groups.py b/nncf/openvino/graph/metatypes/groups.py index 799b6ce92bd..0b94812ab68 100644 --- a/nncf/openvino/graph/metatypes/groups.py +++ b/nncf/openvino/graph/metatypes/groups.py @@ -13,6 +13,7 @@ QUANTIZE_AGNOSTIC_OPERATIONS = [ ov_metatypes.OVMaxPoolMetatype, + ov_metatypes.OVAdaptiveMaxPoolMetatype, ov_metatypes.OVReduceMaxMetatype, ov_metatypes.OVReshapeMetatype, ov_metatypes.OVSqueezeMetatype, @@ -62,6 +63,7 @@ ov_metatypes.OVMaximumMetatype, ov_metatypes.OVMinimumMetatype, ov_metatypes.OVAvgPoolMetatype, + ov_metatypes.OVAdaptiveAvgPoolMetatype, ov_metatypes.OVReduceMeanMetatype, ov_metatypes.OVMVNMetatype, ov_metatypes.OVNormalizeL2Metatype, diff --git a/nncf/openvino/graph/metatypes/openvino_metatypes.py b/nncf/openvino/graph/metatypes/openvino_metatypes.py index 87a2677c0d7..0c7c43f3fac 100644 --- a/nncf/openvino/graph/metatypes/openvino_metatypes.py +++ b/nncf/openvino/graph/metatypes/openvino_metatypes.py @@ -157,6 +157,13 @@ class OVAvgPoolMetatype(OVOpMetatype): hw_config_names = [HWConfigOpName.AVGPOOL] +@OV_OPERATOR_METATYPES.register() +class OVAdaptiveAvgPoolMetatype(OVOpMetatype): + name = "AdaptiveAvgPoolOp" + op_names = ["AdaptiveAvgPool"] + hw_config_names = [HWConfigOpName.AVGPOOL] + + @OV_OPERATOR_METATYPES.register() class OVMaxPoolMetatype(OVOpMetatype): name = "MaxPoolOp" @@ -164,6 +171,13 @@ class OVMaxPoolMetatype(OVOpMetatype): hw_config_names = [HWConfigOpName.MAXPOOL] +@OV_OPERATOR_METATYPES.register() +class OVAdaptiveMaxPoolMetatype(OVOpMetatype): + name = "AdaptiveMaxPoolOp" + op_names = ["AdaptiveMaxPool"] + hw_config_names = [HWConfigOpName.MAXPOOL] + + @OV_OPERATOR_METATYPES.register() class OVConstantMetatype(OVOpMetatype): name = "ConstantOp"