mct-nightly 2.2.0.20241211.531__py3-none-any.whl → 2.2.0.20241213.540__py3-none-any.whl

{mct_nightly-2.2.0.20241211.531.dist-info → mct_nightly-2.2.0.20241213.540.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mct-nightly
-Version: 2.2.0.20241211.531
+Version: 2.2.0.20241213.540
 Summary: A Model Compression Toolkit for neural networks
 Home-page: UNKNOWN
 License: UNKNOWN

{mct_nightly-2.2.0.20241211.531.dist-info → mct_nightly-2.2.0.20241213.540.dist-info}/RECORD

@@ -1,4 +1,4 @@
-model_compression_toolkit/__init__.py,sha256=3AkzGEeakmlxtSQtujDenxH119YlHAatNP77RqE6mR4,1573
+model_compression_toolkit/__init__.py,sha256=CLr_OaMRtKsOBdYNYUxEuaSfCgNbKNXmdcf9Ez6b_7U,1573
 model_compression_toolkit/constants.py,sha256=i_R6uXBfO1ph_X6DNJych2x59SUojfJbn7dNjs_mZnc,3846
 model_compression_toolkit/defaultdict.py,sha256=LSc-sbZYXENMCw3U9F4GiXuv67IKpdn0Qm7Fr11jy-4,2277
 model_compression_toolkit/logger.py,sha256=3DByV41XHRR3kLTJNbpaMmikL8icd9e1N-nkQAY9oDk,4567
@@ -33,7 +33,7 @@ model_compression_toolkit/core/common/fusion/graph_fuser.py,sha256=5VkHB2fW0ohfP
 model_compression_toolkit/core/common/fusion/layer_fusing.py,sha256=lOubqpc18TslhXZijWUJQAa1c3jIB2S-M-5HK78wJPQ,5548
 model_compression_toolkit/core/common/graph/__init__.py,sha256=Xr-Lt_qXMdrCnnOaUS_OJP_3iTTGfPCLf8_vSrQgCs0,773
 model_compression_toolkit/core/common/graph/base_graph.py,sha256=GG13PAtndsMjIqINfrCN6llVkFrg5CBfij4z99ntieU,37815
-model_compression_toolkit/core/common/graph/base_node.py,sha256=mGiDcHnL5KybDYSiONSWtjrHOXI6tjjfACfjv2Piogc,31756
+model_compression_toolkit/core/common/graph/base_node.py,sha256=QPV2PWpYGtS2qB-uioYeGEgEEnPuOy-UFOLVyjWWIvg,31877
 model_compression_toolkit/core/common/graph/edge.py,sha256=buoSEUZwilWBK3WeBKpJ-GeDaUA1SDdOHxDpxU_bGpk,3784
 model_compression_toolkit/core/common/graph/functional_node.py,sha256=GH5wStmw8SoAj5IdT_-ItN1Meo_P5NUTt_5bgJC4fak,3935
 model_compression_toolkit/core/common/graph/graph_matchers.py,sha256=CrDoHYq4iPaflgJWmoJ1K4ziLrRogJvFTVWg8P0UcDU,4744
@@ -111,7 +111,7 @@ model_compression_toolkit/core/common/quantization/quantization_fn_selection.py,
 model_compression_toolkit/core/common/quantization/quantization_params_fn_selection.py,sha256=MwIOBZ4BlZSTIOG75PDvlI3JmZ6t8YjPc1VP9Adei60,3847
 model_compression_toolkit/core/common/quantization/quantize_graph_weights.py,sha256=N005MSvx8UypVpa7XrxNrB2G732n2wHj3RmLyjTgd3I,2728
 model_compression_toolkit/core/common/quantization/quantize_node.py,sha256=cdzGNWfT4MRogIU8ehs0tr3lVjnzAI-jeoS9b4TwVBo,2854
-model_compression_toolkit/core/common/quantization/set_node_quantization_config.py,sha256=h8SwVIwM36nBkqd1i-y1Sn_CrIRRp1JP7B6q0L3AQgc,21136
+model_compression_toolkit/core/common/quantization/set_node_quantization_config.py,sha256=ChhXckn8E3Kot6g5R27KD2WSVMzPfqVwkeTJ0UeBdyg,21257
 model_compression_toolkit/core/common/quantization/quantization_params_generation/__init__.py,sha256=eCDGwsWYLU6z7qbEVb4TozMW_nd5VEP_iCJ6PcvyEPw,1486
 model_compression_toolkit/core/common/quantization/quantization_params_generation/error_functions.py,sha256=fzUvqmXVgzp_IV5ER-20kKzl4m8U_shZsAKs-ehhjFo,23887
 model_compression_toolkit/core/common/quantization/quantization_params_generation/lut_kmeans_params.py,sha256=RL-PklAjGyC-26anSt8fU07a6pB_LBQFQy9o4e9giN0,8739
@@ -432,8 +432,8 @@ model_compression_toolkit/target_platform_capabilities/constants.py,sha256=iJXGy
 model_compression_toolkit/target_platform_capabilities/immutable.py,sha256=YhROBiXEIB3TU-bAFrnL3qbAsb1yuWPBAQ_CLOJbYUU,1827
 model_compression_toolkit/target_platform_capabilities/schema/__init__.py,sha256=pKAdbTCFM_2BrZXUtTIw0ouKotrWwUDF_hP3rPwCM2k,696
 model_compression_toolkit/target_platform_capabilities/schema/mct_current_schema.py,sha256=MxEMpEYHpcgGqHMn371jJU6SMCMvV9qXn1NvjQ4XFik,485
-model_compression_toolkit/target_platform_capabilities/schema/schema_functions.py,sha256=jksWB8EkIOY94z7DVhcaynfp7Jy_aJu5zDFjWMQk8O0,1383
-model_compression_toolkit/target_platform_capabilities/schema/v1.py,sha256=tUvJR1tJpo79WuBya4m2HFHrkyxWwfO-cJxtuRZv4RQ,30110
+model_compression_toolkit/target_platform_capabilities/schema/schema_functions.py,sha256=8Ek7nigw_RYLc5eRjbKr70aqysu_GD_hE1AtcKGi6yg,4610
+model_compression_toolkit/target_platform_capabilities/schema/v1.py,sha256=R_CjTWFiVja11WzOxRNqlOaj36zRJyaAE5s8FaP-Yx4,29147
 model_compression_toolkit/target_platform_capabilities/target_platform/__init__.py,sha256=0l1Y2EqivDkWRRRhirSRI3vwUzhUcSTrYgrwKIqO0IM,1548
 model_compression_toolkit/target_platform_capabilities/target_platform/current_tp_model.py,sha256=1Glr4qKDJfdk5TwM5fzZ12XzgbpQFioDOxb475905gk,2013
 model_compression_toolkit/target_platform_capabilities/target_platform/target_platform_model.py,sha256=TDJ65oCoMCVf2vqOlyioAZeOaPAILhNHcxLVMfszoII,1346
@@ -441,8 +441,8 @@ model_compression_toolkit/target_platform_capabilities/target_platform/targetpla
 model_compression_toolkit/target_platform_capabilities/target_platform/targetplatform2framework/attribute_filter.py,sha256=jfhszvuD2Fyy6W2KjlLzXBQKFzTqGAaDZeFVr4-ONQw,8776
 model_compression_toolkit/target_platform_capabilities/target_platform/targetplatform2framework/current_tpc.py,sha256=fIheShGOnxWYKqT8saHpBJqOU5RG_1Hp9qHry7IviIw,2115
 model_compression_toolkit/target_platform_capabilities/target_platform/targetplatform2framework/layer_filter_params.py,sha256=Cl6-mACpje2jM8RJkibbqE3hvTkFR3r26-lW021mIiA,4019
-model_compression_toolkit/target_platform_capabilities/target_platform/targetplatform2framework/operations_to_layers.py,sha256=Rkk-6p1jyfriGMYjAWnDhwZUqMq_lN2xTUpZ4MzU9zI,6717
-model_compression_toolkit/target_platform_capabilities/target_platform/targetplatform2framework/target_platform_capabilities.py,sha256=qtNVVVzei96dxa9MCBrHhGsE8So01i3E-aCZw_400H0,9676
+model_compression_toolkit/target_platform_capabilities/target_platform/targetplatform2framework/operations_to_layers.py,sha256=YcHsObQ0RFT-ETwRSqhQil3xPwik0euh_Y5V9bpWD8E,6865
+model_compression_toolkit/target_platform_capabilities/target_platform/targetplatform2framework/target_platform_capabilities.py,sha256=jQ7t1wCOa-h_OCYOCmNVhEwftkbU1O4d7bnd4PgbuLI,9865
 model_compression_toolkit/target_platform_capabilities/target_platform/targetplatform2framework/target_platform_capabilities_component.py,sha256=FvrYI0Qy7DCmDp2gyUYyCZq5pY84JgLtJqSIiVTJ8Ss,1030
 model_compression_toolkit/target_platform_capabilities/tpc_models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 model_compression_toolkit/target_platform_capabilities/tpc_models/get_target_platform_capabilities.py,sha256=CWind2Kd91lzBTRAh1A9sHuNw17xXhMb3gV436RpK8c,3033
@@ -450,35 +450,35 @@ model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/__i
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/target_platform_capabilities.py,sha256=yIWwvfTpiT0wRf7GwPgK9elKbGh46jxCrkcLVEjvesU,6081
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/latest/__init__.py,sha256=eVIRpx5O0JQI7TSdw5JAWtwrG3MQ8-7hYThQvB9da5c,1528
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1/__init__.py,sha256=1mMOREEMoNHu_KTMGDp4crN61opKWX6aFn1DrDLvqcc,717
-model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1/tp_model.py,sha256=-5fj8VUGYGOWpvN_RsBRPA6bYeW7ncH0S8Atbn1e8xI,11343
+model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1/tp_model.py,sha256=W-DxFuYuD6cclQHmHmVdUpILHqTL7VfOKFFosoUtcJw,11317
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1/tpc_keras.py,sha256=DiEVKde6bZiRDcoeTKzbrfjFIDVEeES0Q_Ms50NbW7Y,6568
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1/tpc_pytorch.py,sha256=rQY-ApvLKo9LbN2uKO2NsdOgX49TKcED3o8mBxWAFKw,5594
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1_lut/__init__.py,sha256=vFDyiMymNZSRCdTgAyWn4A-tZD3vzze_PTLBSF2OYe8,721
-model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1_lut/tp_model.py,sha256=MIkzWvregjIPUDJpt9YCzQdH5krDKt0uYq1C1xszKr4,11066
+model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1_lut/tp_model.py,sha256=ArqqPg3cxCuRFQzz5lGMrZcLxpx_8n8_6cdFzKNTtDU,11072
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1_lut/tpc_keras.py,sha256=I3NnmCBSB6KB8ECXJsfXw6FD-ONj39RXL-yB2-nE1xI,6577
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1_lut/tpc_pytorch.py,sha256=R6ZDXT2bwAmdZfmXxIoCo7-_h7oGfyr6cZoZmBJdPa0,5713
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1_pot/__init__.py,sha256=NUuczImqUxzdfflqSdqkeAN8aCU6Tuiu6U0Fnj9Tzmw,721
-model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1_pot/tp_model.py,sha256=QHVG6aQLWBT60hdrCvbOfSZsGYD-1SKHR732TMYfNbI,10821
+model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1_pot/tp_model.py,sha256=fwbZoIOrQhPXbcC8kAo3Gl_hmLTbr7a2BxiRvB1DxV8,10827
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1_pot/tpc_keras.py,sha256=ovHI4lvlcP_haolXmL8TvIpPi9tglhdUxRxNxuf6QPk,6581
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1_pot/tpc_pytorch.py,sha256=PgNAmXZx7hg46KwR16laU7jr2U35rnHiTcR25RaLxd0,5732
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v2/__init__.py,sha256=vKWAoQ2KkhuptS5HZB50zHG6KY8wHpHTxPugw_nGCRo,717
-model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v2/tp_model.py,sha256=hme-SsuB8pZh9OPIRPujjJZ24EoU0y_5rrgNBoc2osY,11398
+model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v2/tp_model.py,sha256=3qQ-RQPFjTcE8WM3rzMYe89SFoMtkbJ1f6IjkWUam3Q,11372
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v2/tpc_keras.py,sha256=i6oEt_3FAeccU9QkVCVxWZf2aX17IExAljR8y13k9P4,6663
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v2/tpc_pytorch.py,sha256=Iwi-D_0iC19Lo3yWduuaItezXFrRxRhowUY4jGOtwXk,5700
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v2_lut/__init__.py,sha256=wUk4Xsg7jpxOWYjq2K3WUwLcI185p_sVPK-ttG0ydhA,721
-model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v2_lut/tp_model.py,sha256=xQHVTmpGRImcnqhQ6AdmB8JSWiCLydmAWRZyD4U3TIk,11118
+model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v2_lut/tp_model.py,sha256=yANACBcyagUsJ3D8oustmUCVNrKcwFlBMAMTCrQtac4,11124
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v2_lut/tpc_keras.py,sha256=e3su6KmNsc2ziP3eEICp-tcnY9YmGPiLoMJ3TreqVx4,6671
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v2_lut/tpc_pytorch.py,sha256=9W-pjOhONqgym1t2vDRTo7x-JQiVlxQsAKAG6HdAww4,5707
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v3/__init__.py,sha256=gAeebYCKyIXH9-Qwze7FwvTihudzAHk_Qsg94fQbkjQ,717
-model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v3/tp_model.py,sha256=4iztway_Dd6igh3qY--d2AYPjWSorhO03Tdi_Mcm-yE,13210
+model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v3/tp_model.py,sha256=acSYyoS6R6DVHtyExcj93zHz5g3YKiZ71dF6f7N3ygQ,13184
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v3/tpc_keras.py,sha256=vbBgQfEd5mLnJKGX2CO-KA0x87_qTmDH7HbbtLgCIfg,6881
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v3/tpc_pytorch.py,sha256=oDG_Ln0mPmt8p5eKrjGQlHLHCApErLHazny9bIfu3iI,5968
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v3_lut/__init__.py,sha256=C2kwyDE1-rtukkbNSoKRv9q8Nt2GOCaBbl0BdOr3goA,721
-model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v3_lut/tp_model.py,sha256=LOQgPS0yLlrZK83R1ZRr6ZZwo44eojGZ7fbtH9g_5JA,12008
+model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v3_lut/tp_model.py,sha256=QdicH7CpS7-wUdp4XnYLL9yGmly_RKPifB4sNf1pUNc,12014
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v3_lut/tpc_keras.py,sha256=hHK7MY6xfSCRnXQ6Qz0hbN_hM18f-gf3uZmxdl44khU,6679
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v3_lut/tpc_pytorch.py,sha256=GmS3CuSQPwLSgPdruXCSHN49_fRiRBn9PKA9CPmtqM8,5789
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v4/__init__.py,sha256=tHTUvsaerSfbe22pU0kIDauPpFD7Pq5EmZytVIDkHz4,717
-model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v4/tp_model.py,sha256=jATi0V3ZhgFVfMag2l4MElc3e5XN8eBBytWUxoEj4nU,17532
+model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v4/tp_model.py,sha256=4Gjb_6pSgnCi7xqUYKdUsBBZeRNcrBbQ7wvuprHHIms,17498
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v4/tpc_keras.py,sha256=9LToJzBBCXEZ5AOj2zzJMRr6S0uwlKKEg3stLK2u1fw,8341
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v4/tpc_pytorch.py,sha256=v1Fyp9YKWfdxF9tLfD0fAV41mqEH0S9z-jqr4QFH5CQ,7221
 model_compression_toolkit/target_platform_capabilities/tpc_models/qnnpack_tpc/__init__.py,sha256=cco4TmeIDIh32nj9ZZXVkws4dd9F2UDrmjKzTN8G0V0,697
@@ -492,7 +492,7 @@ model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/__i
 model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/target_platform_capabilities.py,sha256=_tf41m40fbax27y5A5JoGHw4p5NY-Kb3c8oxSTnRD_E,2802
 model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/latest/__init__.py,sha256=LIUUQn42YU7oD2YfnEgP0gfqm7Hq9e0fD_8418aKzKI,1511
 model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/v1/__init__.py,sha256=t4JKsPcor-7KSCKzIwuaBv0NLNwfhuewAQGlDl6iBeo,717
-model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/v1/tp_model.py,sha256=4rod_xFp56WoRIeLF6cxZWGniE0vtyqvdSdXaA0GsNc,10385
+model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/v1/tp_model.py,sha256=D1aMSpT4E-WJ4V3rPfjbl_FNZTuFI3-AGsQiL4-6ujc,10389
 model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/v1/tpc_keras.py,sha256=4vQG9uaV5bAte1VP8oMT5dwiDisW84F1B-de0BdBusY,6808
 model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/v1/tpc_pytorch.py,sha256=uG27HAOrD-JMSDXaEycDJeFGqhF8J3ZqKI4CJCW6zA4,5934
 model_compression_toolkit/trainable_infrastructure/__init__.py,sha256=uewpvlPkH9mBFt8IxoAgIfz6iEcvWbOImm_fb6_BxD8,1543
@@ -559,8 +559,8 @@ model_compression_toolkit/xquant/pytorch/model_analyzer.py,sha256=b93o800yVB3Z-i
 model_compression_toolkit/xquant/pytorch/pytorch_report_utils.py,sha256=bOc-hFL3gdoSM1Th_S2N_-9JJSlPGpZCTx_QLJHS6lg,3388
 model_compression_toolkit/xquant/pytorch/similarity_functions.py,sha256=CERxq5K8rqaiE-DlwhZBTUd9x69dtYJlkHOPLB54vm8,2354
 model_compression_toolkit/xquant/pytorch/tensorboard_utils.py,sha256=mkoEktLFFHtEKzzFRn_jCnxjhJolK12TZ5AQeDHzUO8,9767
-mct_nightly-2.2.0.20241211.531.dist-info/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
-mct_nightly-2.2.0.20241211.531.dist-info/METADATA,sha256=qdiuketUM1fffHOn2tTgXRwUHPHj-IcdJbapmxVE0bM,26446
-mct_nightly-2.2.0.20241211.531.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-mct_nightly-2.2.0.20241211.531.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
-mct_nightly-2.2.0.20241211.531.dist-info/RECORD,,
+mct_nightly-2.2.0.20241213.540.dist-info/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
+mct_nightly-2.2.0.20241213.540.dist-info/METADATA,sha256=_PxYI2q2aQe9fwkTQJ0ec5vjTiBG1lktdkehLb4p7YI,26446
+mct_nightly-2.2.0.20241213.540.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+mct_nightly-2.2.0.20241213.540.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
+mct_nightly-2.2.0.20241213.540.dist-info/RECORD,,

model_compression_toolkit/__init__.py

@@ -27,4 +27,4 @@ from model_compression_toolkit import data_generation
 from model_compression_toolkit import pruning
 from model_compression_toolkit.trainable_infrastructure.keras.load_model import keras_load_quantized_model
 
-__version__ = "2.2.0.20241211.000531"
+__version__ = "2.2.0.20241213.000540"

model_compression_toolkit/core/common/graph/base_node.py

@@ -24,6 +24,7 @@ from model_compression_toolkit.core.common.quantization.node_quantization_config
 from model_compression_toolkit.logger import Logger
 from model_compression_toolkit.target_platform_capabilities.schema.mct_current_schema import QuantizationConfigOptions, \
     OpQuantizationConfig
+from model_compression_toolkit.target_platform_capabilities.schema.schema_functions import max_input_activation_n_bits
 from model_compression_toolkit.target_platform_capabilities.target_platform import TargetPlatformCapabilities, LayerFilterParams
 
 
@@ -584,7 +585,7 @@ class BaseNode:
         _node_qc_options = node_qc_options.quantization_config_list
         if len(next_nodes):
             next_nodes_qc_options = [_node.get_qco(tpc) for _node in next_nodes]
-            next_nodes_supported_input_bitwidth = min([op_cfg.max_input_activation_n_bits
+            next_nodes_supported_input_bitwidth = min([max_input_activation_n_bits(op_cfg)
                                                        for qc_opts in next_nodes_qc_options
                                                        for op_cfg in qc_opts.quantization_config_list])
 
@@ -595,7 +596,7 @@ class BaseNode:
                 Logger.critical(f"Graph doesn't match TPC bit configurations: {self} -> {next_nodes}.")  # pragma: no cover
 
             # Verify base config match
-            if any([node_qc_options.base_config.activation_n_bits > qc_opt.base_config.max_input_activation_n_bits
+            if any([node_qc_options.base_config.activation_n_bits > max_input_activation_n_bits(qc_opt.base_config)
                     for qc_opt in next_nodes_qc_options]):
                 # base_config activation bits doesn't match next node supported input bit-width -> replace with
                 # a qco from quantization_config_list with maximum activation bit-width.

model_compression_toolkit/core/common/quantization/set_node_quantization_config.py

@@ -32,6 +32,7 @@ from model_compression_toolkit.core.common.quantization.quantization_params_fn_s
     get_activation_quantization_params_fn, get_weights_quantization_params_fn
 from model_compression_toolkit.core.common.quantization.quantization_fn_selection import \
     get_weights_quantization_fn
+from model_compression_toolkit.target_platform_capabilities.schema.schema_functions import max_input_activation_n_bits
 from model_compression_toolkit.target_platform_capabilities.target_platform.targetplatform2framework import TargetPlatformCapabilities
 from model_compression_toolkit.target_platform_capabilities.schema.mct_current_schema import OpQuantizationConfig, \
     QuantizationConfigOptions
@@ -117,7 +118,7 @@ def filter_node_qco_by_graph(node: BaseNode,
 
     if len(next_nodes):
         next_nodes_qc_options = [_node.get_qco(tpc) for _node in next_nodes]
-        next_nodes_supported_input_bitwidth = min([op_cfg.max_input_activation_n_bits
+        next_nodes_supported_input_bitwidth = min([max_input_activation_n_bits(op_cfg)
                                                    for qc_opts in next_nodes_qc_options
                                                    for op_cfg in qc_opts.quantization_config_list])
 
@@ -128,7 +129,7 @@ def filter_node_qco_by_graph(node: BaseNode,
             Logger.critical(f"Graph doesn't match TPC bit configurations: {node} -> {next_nodes}.")
 
         # Verify base config match
-        if any([node_qc_options.base_config.activation_n_bits > qc_opt.base_config.max_input_activation_n_bits
+        if any([node_qc_options.base_config.activation_n_bits > max_input_activation_n_bits(qc_opt.base_config)
                 for qc_opt in next_nodes_qc_options]):
             # base_config activation bits doesn't match next node supported input bit-width -> replace with
             # a qco from quantization_config_list with maximum activation bit-width.

model_compression_toolkit/target_platform_capabilities/schema/schema_functions.py

@@ -12,26 +12,95 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-import copy
-from typing import Any, Dict
+from logging import Logger
+from typing import Optional
 
+from model_compression_toolkit.target_platform_capabilities.schema.mct_current_schema import OpQuantizationConfig, \
+    TargetPlatformModel, QuantizationConfigOptions, OperatorsSetBase
 
-def clone_and_edit_object_params(obj: Any, **kwargs: Dict) -> Any:
+
+def max_input_activation_n_bits(op_quantization_config: OpQuantizationConfig) -> int:
+    """
+    Get the maximum supported input bit-width.
+
+    Args:
+        op_quantization_config (OpQuantizationConfig):  The configuration object from which to retrieve the maximum supported input bit-width.
+
+    Returns:
+        int: Maximum supported input bit-width.
+    """
+    return max(op_quantization_config.supported_input_activation_n_bits)
+
+
+def get_config_options_by_operators_set(tp_model: TargetPlatformModel,
+                                        operators_set_name: str) -> QuantizationConfigOptions:
+    """
+    Get the QuantizationConfigOptions of an OperatorsSet by its name.
+
+    Args:
+        tp_model (TargetPlatformModel): The target platform model containing the operator sets and their configurations.
+        operators_set_name (str): The name of the OperatorsSet whose quantization configuration options are to be retrieved.
+
+    Returns:
+        QuantizationConfigOptions: The quantization configuration options associated with the specified OperatorsSet,
+        or the default quantization configuration options if the OperatorsSet is not found.
+    """
+    for op_set in tp_model.operator_set:
+        if operators_set_name == op_set.name:
+            return op_set.qc_options
+    return tp_model.default_qco
+
+
+def get_default_op_quantization_config(tp_model: TargetPlatformModel) -> OpQuantizationConfig:
     """
-    Clones the given object and edit some of its parameters.
+    Get the default OpQuantizationConfig of the TargetPlatformModel.
 
     Args:
-        obj: An object to clone.
-        **kwargs: Keyword arguments to edit in the cloned object.
+        tp_model (TargetPlatformModel): The target platform model containing the default quantization configuration.
 
     Returns:
-        Edited copy of the given object.
+        OpQuantizationConfig: The default quantization configuration.
+
+    Raises:
+        AssertionError: If the default quantization configuration list contains more than one configuration option.
+    """
+    assert len(tp_model.default_qco.quantization_config_list) == 1, \
+        f"Default quantization configuration options must contain only one option, " \
+        f"but found {len(tp_model.default_qco.quantization_config_list)} configurations." # pragma: no cover
+    return tp_model.default_qco.quantization_config_list[0]
+
+
+def is_opset_in_model(tp_model: TargetPlatformModel, opset_name: str) -> bool:
+    """
+    Check whether an OperatorsSet is defined in the model.
+
+    Args:
+        tp_model (TargetPlatformModel): The target platform model containing the list of operator sets.
+        opset_name (str): The name of the OperatorsSet to check for existence.
+
+    Returns:
+        bool: True if an OperatorsSet with the given name exists in the target platform model,
+              otherwise False.
     """
+    return opset_name in [x.name for x in tp_model.operator_set]
 
-    obj_copy = copy.deepcopy(obj)
-    for k, v in kwargs.items():
-        assert hasattr(obj_copy,
-                       k), f'Edit parameter is possible only for existing parameters in the given object, ' \
-                           f'but {k} is not a parameter of {obj_copy}.'
-        setattr(obj_copy, k, v)
-    return obj_copy
+
+def get_opset_by_name(tp_model: TargetPlatformModel, opset_name: str) -> Optional[OperatorsSetBase]:
+    """
+    Get an OperatorsSet object from the model by its name.
+
+    Args:
+        tp_model (TargetPlatformModel): The target platform model containing the list of operator sets.
+        opset_name (str): The name of the OperatorsSet to be retrieved.
+
+    Returns:
+        Optional[OperatorsSetBase]: The OperatorsSet object with the specified name if found.
+        If no operator set with the specified name is found, None is returned.
+
+    Raises:
+        A critical log message if multiple operator sets with the same name are found.
+    """
+    opset_list = [x for x in tp_model.operator_set if x.name == opset_name]
+    if len(opset_list) > 1:
+        Logger.critical(f"Found more than one OperatorsSet in TargetPlatformModel with the name {opset_name}.") # pragma: no cover
+    return opset_list[0] if opset_list else None