mct-nightly 2.4.0.20250706.701__py3-none-any.whl → 2.4.0.20250708.612__py3-none-any.whl

{mct_nightly-2.4.0.20250706.701.dist-info → mct_nightly-2.4.0.20250708.612.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mct-nightly
-Version: 2.4.0.20250706.701
+Version: 2.4.0.20250708.612
 Summary: A Model Compression Toolkit for neural networks
 Author-email: ssi-dnn-dev@sony.com
 Classifier: Programming Language :: Python :: 3

{mct_nightly-2.4.0.20250706.701.dist-info → mct_nightly-2.4.0.20250708.612.dist-info}/RECORD

@@ -1,5 +1,5 @@
-mct_nightly-2.4.0.20250706.701.dist-info/licenses/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
-model_compression_toolkit/__init__.py,sha256=oMuUjzzPKDPdfhr1AF-A55kyQJBPEDqlrqfoPxw3HKA,1557
+mct_nightly-2.4.0.20250708.612.dist-info/licenses/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
+model_compression_toolkit/__init__.py,sha256=TrwEn4n4YKNNqTg96Ud45uv2qh_Ob-Q1IzRMrsqjJG4,1557
 model_compression_toolkit/constants.py,sha256=KNgiNLpsMgSYyXMNEbHXd4bFNerQc1D6HH3vpbUq_Gs,4086
 model_compression_toolkit/defaultdict.py,sha256=LSc-sbZYXENMCw3U9F4GiXuv67IKpdn0Qm7Fr11jy-4,2277
 model_compression_toolkit/logger.py,sha256=L3q7tn3Uht0i_7phnlOWMR2Te2zvzrt2HOz9vYEInts,4529
@@ -7,8 +7,8 @@ model_compression_toolkit/metadata.py,sha256=x_Bk4VpzILdsFax6--CZ3X18qUTP28sbF_A
 model_compression_toolkit/verify_packages.py,sha256=l0neIRr8q_QwxmuiTI4vyCMDISDedK0EihjEQUe66tE,1319
 model_compression_toolkit/core/__init__.py,sha256=HNverPpoqEyFKTa7iEdOqqY2P0Gq-7GMejNOi6ZPcQs,2042
 model_compression_toolkit/core/analyzer.py,sha256=5P03LbkFy-mu31TMAiQoIKcsA1-DNz7cTzkGvRaXtbw,3505
-model_compression_toolkit/core/graph_prep_runner.py,sha256=naZWayASraZ9PgmqCBFgFWWfDV3zLgPaIo6JLbInZc4,11361
-model_compression_toolkit/core/quantization_prep_runner.py,sha256=tz91E1BaNc_K0lvVZGB8oS6ya5N4Z5TJLG4pSM3hx30,6229
+model_compression_toolkit/core/graph_prep_runner.py,sha256=XvuR1lNsbaGG3HyROA8nF2n0oPNb1Tnw6rGatewAvE0,10563
+model_compression_toolkit/core/quantization_prep_runner.py,sha256=W6CAJ_Euxrjm5jj__QqfB4xsD8NYx_-XWX22h0ax6B4,6262
 model_compression_toolkit/core/runner.py,sha256=QpiJQmQXK6mWmnygNRdy6I8S48DHV-B0Kmr4TqOKbeA,12418
 model_compression_toolkit/core/common/__init__.py,sha256=Wh127PbXcETZX_d1PQqZ71ETK3J9XO5A-HpadGUbj6o,1447
 model_compression_toolkit/core/common/base_substitutions.py,sha256=xDFSmVVs_iFSZfajytI0cuQaNRNcwHX3uqOoHgVUvxQ,1666
@@ -16,18 +16,17 @@ model_compression_toolkit/core/common/framework_implementation.py,sha256=jrTupZb
 model_compression_toolkit/core/common/framework_info.py,sha256=vPGV28gm-kvNSkkWI6jY3YeKBUYmn6UQ98HVUnl_-tM,5449
 model_compression_toolkit/core/common/memory_computation.py,sha256=ixoSpV5ZYZGyzhre3kQcvR2sNA8KBsPZ3lgbkDnw9Cs,1205
 model_compression_toolkit/core/common/model_builder_mode.py,sha256=jll9-59OPaE3ug7Y9-lLyV99_FoNHxkGZMgcm0Vkpss,1324
-model_compression_toolkit/core/common/model_collector.py,sha256=A1uaGmxqj-392lMtE-F020FHFAyyKDJDdeJeZYtkv3Y,12755
-model_compression_toolkit/core/common/model_validation.py,sha256=HRnYh2uY85yJ7Ijmt4tKRn8bMg60zbBSDRCgK246gUM,1067
+model_compression_toolkit/core/common/model_collector.py,sha256=I0O5SoUrwB045AGLleOGYVvZyR7HJx_t6vfbECahfoU,13914
 model_compression_toolkit/core/common/node_prior_info.py,sha256=WXX_PrGVG9M9I_REG5ZzFBohwmV4yf356sZnrja_FLo,2832
 model_compression_toolkit/core/common/similarity_analyzer.py,sha256=S3f6WgHyw62dGcxpX51FGKyfebe2zv9ABKbjtGyKRvY,9215
 model_compression_toolkit/core/common/user_info.py,sha256=dSRMnT-oewmdOziIpEuW-s9K7vTSeyUBxT4z9neXurI,1648
 model_compression_toolkit/core/common/back2framework/__init__.py,sha256=cco4TmeIDIh32nj9ZZXVkws4dd9F2UDrmjKzTN8G0V0,697
 model_compression_toolkit/core/common/back2framework/base_model_builder.py,sha256=yrIxT0ttDi9XViy8Zt8apnMCT8xDyVd5HZp0IttrGGQ,1775
 model_compression_toolkit/core/common/collectors/__init__.py,sha256=sw7LOPN1bM82o3SkMaklyH0jw-TLGK0-fl2Wq73rffI,697
-model_compression_toolkit/core/common/collectors/base_collector.py,sha256=JoBTX3rRcRnUF3_Azjg848aiJt9drCJ5TsR9RahVI0Y,2591
+model_compression_toolkit/core/common/collectors/base_collector.py,sha256=n3IIwfWVIl8TO_7MR9jBBxzf2foysWMyrGS-oaCuKiA,2664
 model_compression_toolkit/core/common/collectors/histogram_collector.py,sha256=zra5V06Brpjc1cUNIMVVGqdoqAuro62_hGy2Zm5-XMQ,6754
-model_compression_toolkit/core/common/collectors/mean_collector.py,sha256=mjr3U_z7vn8rrqpkHnfErUOflToIYl4ozBVzP2awqDQ,3414
-model_compression_toolkit/core/common/collectors/min_max_per_channel_collector.py,sha256=5oKsJEKdVmj4C7fKdHhmrFN5k4G2BaFETpmf_xKNs7s,5207
+model_compression_toolkit/core/common/collectors/mean_collector.py,sha256=7-oCVU76e9ZWIZLiLte-z07FX2Rjmu5Go-KRj7b7oWY,3564
+model_compression_toolkit/core/common/collectors/min_max_per_channel_collector.py,sha256=pAdiLXNNm5sfIZ584m2MzJQ0PTNp6MKsrMxJ_onEN1s,5314
 model_compression_toolkit/core/common/collectors/statistics_collector.py,sha256=psijsQZefwjMDH8SU5E18n65HiGtQilPhKr1hhzZX-I,8268
 model_compression_toolkit/core/common/collectors/weighted_histogram_collector.py,sha256=zp3dE7YTqWmkD5QWdRhsl9zD8W6Lr96G1Wjw1g2D3T0,4894
 model_compression_toolkit/core/common/fusion/__init__.py,sha256=Rf1RcYmelmdZmBV5qOKvKWF575ofc06JFQSq83Jz99A,696
@@ -87,7 +86,7 @@ model_compression_toolkit/core/common/network_editors/node_filters.py,sha256=Pc_
 model_compression_toolkit/core/common/pruning/__init__.py,sha256=DGJybkDQtKMSMFoZ-nZ3ZifA8uJ6G_D20wHhKHNlmU0,699
 model_compression_toolkit/core/common/pruning/channels_grouping.py,sha256=-zrq0TsfVE4ooxOcJCsL8H2DBau6vSkEKz1ot-x-Faw,3736
 model_compression_toolkit/core/common/pruning/greedy_mask_calculator.py,sha256=UZekrges7gZv17JFLX_AV2Kv0eBXXarMNInzuOTlyvA,7712
-model_compression_toolkit/core/common/pruning/memory_calculator.py,sha256=bQtkMuxm9RajIztN4m88ZT0zCeN_bRcm4H2VGqE36lg,18944
+model_compression_toolkit/core/common/pruning/memory_calculator.py,sha256=7IdX9APlS17omWeF_y0ZFr8S2K7Cp-M0u8rFtfmrx8U,19023
 model_compression_toolkit/core/common/pruning/prune_graph.py,sha256=eGvuqrxyADRSvhKz0R_7lLfIl7bKnn4bryElu3LsVcA,3158
 model_compression_toolkit/core/common/pruning/pruner.py,sha256=Zl0IK0anorzagaSP8qXMN31Dtw5m-Le-JRy2baPLs6M,7262
 model_compression_toolkit/core/common/pruning/pruning_config.py,sha256=fbqERt11FGVeuqPVA6nVbgGDh6Ox9mpEKdxVJT8eG4I,3681
@@ -107,8 +106,8 @@ model_compression_toolkit/core/common/quantization/candidate_node_quantization_c
 model_compression_toolkit/core/common/quantization/core_config.py,sha256=yxCzWqldcHoe8GGxrH0tp99bhrc5jDT7SgZftnMUUBE,2374
 model_compression_toolkit/core/common/quantization/debug_config.py,sha256=uH45Uq3Tp9FIyMynex_WY2_y-Kv8LuPw2XXZydnpW5A,1649
 model_compression_toolkit/core/common/quantization/filter_nodes_candidates.py,sha256=FyYCYbfkAofEWO2mAvFIppPeq2I10f1ScPNiVa9F7x4,7687
-model_compression_toolkit/core/common/quantization/node_quantization_config.py,sha256=fj1ebZgnK6xH-9LIAu93rOEU7siXK86U_VyAtUwu9nA,24869
-model_compression_toolkit/core/common/quantization/quantization_config.py,sha256=UkSVW7d1OF_Px9gAjsqqK65aYhIBFWaBO-_IH6_AFfg,4403
+model_compression_toolkit/core/common/quantization/node_quantization_config.py,sha256=VRY_HxZl9D77eq6oJ61eBnL4QJTG5pVLzJtAcaZATRQ,21636
+model_compression_toolkit/core/common/quantization/quantization_config.py,sha256=EMhXeY1qkvwlMAY5fpKRDuLEAyRY5yKqe2fOmAD_rVI,4362
 model_compression_toolkit/core/common/quantization/quantization_fn_selection.py,sha256=VVq2cKjumlNWucUbaNw8s2J0IbI_vrQ-KR_eQPshGSg,3140
 model_compression_toolkit/core/common/quantization/quantize_graph_weights.py,sha256=N005MSvx8UypVpa7XrxNrB2G732n2wHj3RmLyjTgd3I,2728
 model_compression_toolkit/core/common/quantization/quantize_node.py,sha256=WJ-lsT_R_pqjbrMzgcposugACDNz7yZ09vSlltTb78A,3001
@@ -118,10 +117,10 @@ model_compression_toolkit/core/common/quantization/quantization_params_generatio
 model_compression_toolkit/core/common/quantization/quantization_params_generation/lut_kmeans_params.py,sha256=RL-PklAjGyC-26anSt8fU07a6pB_LBQFQy9o4e9giN0,8739
 model_compression_toolkit/core/common/quantization/quantization_params_generation/outlier_filter.py,sha256=9gnfJV89jpGwAx8ImJ5E9NjCv3lDtbyulP4OtgWb62M,1772
 model_compression_toolkit/core/common/quantization/quantization_params_generation/power_of_two_selection.py,sha256=-cghHF5S11qbjTDRruHlc__uaDoofZHl7QTl8hCeKW0,11141
-model_compression_toolkit/core/common/quantization/quantization_params_generation/qparams_activations_computation.py,sha256=3EAbtLHOgTJIMbGlfAzeki7xxjipAsMyAaVRFXqF228,7243
-model_compression_toolkit/core/common/quantization/quantization_params_generation/qparams_computation.py,sha256=27We8-tLL0dkDPYSDlhXe6ZKSO-kw2s5sD4q9I_ADmE,8401
+model_compression_toolkit/core/common/quantization/quantization_params_generation/qparams_activations_computation.py,sha256=vOFquQ3_h0jyG8EEeHM8M57uIKMZQ7iobUgbvWbiXh4,7798
+model_compression_toolkit/core/common/quantization/quantization_params_generation/qparams_computation.py,sha256=TPSkKujdJ6jB7VOVp0kRgMnthmeuaBZgN1HQuJ2pqR0,7951
 model_compression_toolkit/core/common/quantization/quantization_params_generation/qparams_search.py,sha256=Nv_b3DECVjQnlrUet2kbuSvSKVnxcc-gf2zhFb2jSZk,43482
-model_compression_toolkit/core/common/quantization/quantization_params_generation/qparams_weights_computation.py,sha256=jb9Q2WgjmMc6i8j3TXr850tWCdI0a8598bkTmMYfdAY,4529
+model_compression_toolkit/core/common/quantization/quantization_params_generation/qparams_weights_computation.py,sha256=saOQMtj1qYqgQoAFjq31p7xtiRDxmanGGdm0DE81_cg,4820
 model_compression_toolkit/core/common/quantization/quantization_params_generation/symmetric_selection.py,sha256=6tRNgWvn-4r8hiSHqND7Qms1Nje1DUR4MR0JeWCNyvI,12531
 model_compression_toolkit/core/common/quantization/quantization_params_generation/uniform_selection.py,sha256=xiZgCkoIrJ9xsR17x9pSl_sUbiuSta67kf7bQ4quFUI,10804
 model_compression_toolkit/core/common/quantization/quantizers/__init__.py,sha256=mjbqLD-KcG3eNeCYpu1GBS7VclGVOQ63x2p6mAAuba4,698
@@ -130,22 +129,22 @@ model_compression_toolkit/core/common/quantization/quantizers/quantizers_helpers
 model_compression_toolkit/core/common/quantization/quantizers/uniform_quantizers.py,sha256=wXExWHf5-0He7L4bpvFpKlx7FG4u3DAfNZiXPpOs_SQ,5521
 model_compression_toolkit/core/common/statistics_correction/__init__.py,sha256=sw7LOPN1bM82o3SkMaklyH0jw-TLGK0-fl2Wq73rffI,697
 model_compression_toolkit/core/common/statistics_correction/apply_activation_bias_correction_to_graph.py,sha256=oUa1Gv9jIICOoFljTiIaItFjJQPht7CBe-wEr3iBuLQ,4118
-model_compression_toolkit/core/common/statistics_correction/apply_bias_correction_to_graph.py,sha256=eGd0gaPz1K9tzfQf1UMBeshoydFwwZ4Ha2JKFCJ2eZc,4474
-model_compression_toolkit/core/common/statistics_correction/apply_second_moment_correction_to_graph.py,sha256=w9VkX0_XyE64zaYJrZqGEtVxaox7MwY-c8Ie1C0f6ZU,5093
+model_compression_toolkit/core/common/statistics_correction/apply_bias_correction_to_graph.py,sha256=fvvIH5uWtQbmPJQh_zjE-XuCFP9YrlOEs3mW9ysKhms,3652
+model_compression_toolkit/core/common/statistics_correction/apply_second_moment_correction_to_graph.py,sha256=7HIoey1tqbTy1tnTF_KF5D-dSNankamBj8kWO9kUQYo,5200
 model_compression_toolkit/core/common/statistics_correction/compute_activation_bias_correction_of_graph.py,sha256=289b2iwzp2hjsgpEZotQKNB2aPKjAZopRaGnbzErHV8,9263
-model_compression_toolkit/core/common/statistics_correction/compute_bias_correction_of_graph.py,sha256=08k7sqOLIya7Vvg2WMFdaSzLJ2FsgQlcKk0H_KoFoUg,10068
-model_compression_toolkit/core/common/statistics_correction/statistics_correction.py,sha256=yB5Kxk74RAzcXxguFRVpvjFSWFrGrqL3JoU1qLst4PQ,5881
+model_compression_toolkit/core/common/statistics_correction/compute_bias_correction_of_graph.py,sha256=p4u50-1CPuGTQtMDz5uaZqFSLhl2BLTbCLDQvplCuW4,9552
+model_compression_toolkit/core/common/statistics_correction/statistics_correction.py,sha256=F0xOl97p8Q0h4MXxOKo0kGY4iJIiJou06JbICatg8vQ,5881
 model_compression_toolkit/core/common/substitutions/__init__.py,sha256=sw7LOPN1bM82o3SkMaklyH0jw-TLGK0-fl2Wq73rffI,697
 model_compression_toolkit/core/common/substitutions/apply_substitutions.py,sha256=k-bifmakHIYZeZS-4T1QpZ1Et6AwAijMRgAKs7hmMKc,1390
 model_compression_toolkit/core/common/substitutions/batchnorm_folding.py,sha256=wLlTT7sqUffKHwOrMG2VV5SktQkkP54l8taW1Fq0mh0,13392
-model_compression_toolkit/core/common/substitutions/batchnorm_reconstruction.py,sha256=Qe-MYKL2GRQ3PX1Q-zpws5mEW3vrs2h19kjiUZTkKwI,8327
+model_compression_toolkit/core/common/substitutions/batchnorm_reconstruction.py,sha256=xa-6ACozqTu4xWXkC-7p6YTZLcLcH_AUYEROTl8USWc,8014
 model_compression_toolkit/core/common/substitutions/batchnorm_refusing.py,sha256=eCbhbAzgXWoVymMLbrupJ1qAcdhZDwkjKeja0fCymnY,9746
 model_compression_toolkit/core/common/substitutions/linear_collapsing.py,sha256=iEtzbWCDXP6EDkTZCtREQ0rpMxhQ2kM9zlcP_0KLq9I,12367
 model_compression_toolkit/core/common/substitutions/linear_collapsing_substitution.py,sha256=uoauhmncQqUBNvD-qCLIXsIbl_IzrbxSKdxiMig-5W4,2406
 model_compression_toolkit/core/common/substitutions/remove_identity.py,sha256=TKU1TIU52UIkVnl0EZvWnDhLV9nIVZ4hqi-w1i4NXMk,2637
 model_compression_toolkit/core/common/substitutions/residual_collapsing.py,sha256=N82mso5j3EJQlKt9EMHjjEJ67FmdGQeCfN8U5grOFXo,4830
 model_compression_toolkit/core/common/substitutions/scale_equalization.py,sha256=2_NmmBmUBZZwXuF5Od2S919_FgQKYIf-nSyNPawr0e4,9840
-model_compression_toolkit/core/common/substitutions/shift_negative_activation.py,sha256=Q9dQPLIKVtCp23yj-BmQmYkH94OBvAfV-19CYgqWSw0,32572
+model_compression_toolkit/core/common/substitutions/shift_negative_activation.py,sha256=_TliV1tBxizPsCZVbOhDhg3mSc_m0sY9gVqQMYTwQoc,32739
 model_compression_toolkit/core/common/substitutions/softmax_shift.py,sha256=R-0ZqhYAuZLEFWHvB2UTPm52L6gWHGdRdEnwGxKSeGI,2625
 model_compression_toolkit/core/common/substitutions/virtual_activation_weights_composition.py,sha256=cokiYPZB7504oHTlgZy8u2Xv_S-RK_oDSnGvYRX3JK4,4136
 model_compression_toolkit/core/common/substitutions/weights_activation_split.py,sha256=vafrJ6eA37PrIzOs7uOsiJKIBmAVmNJ-wXsoe332BIw,4683
@@ -157,9 +156,8 @@ model_compression_toolkit/core/keras/__init__.py,sha256=mjbqLD-KcG3eNeCYpu1GBS7V
 model_compression_toolkit/core/keras/constants.py,sha256=dh4elQWt6Q6NYRht5k5RiiOcnLAq1v0MMBCJqMJzzFk,3225
 model_compression_toolkit/core/keras/custom_layer_validation.py,sha256=f-b14wuiIgitBe7d0MmofYhDCTO3IhwJgwrh-Hq_t_U,1192
 model_compression_toolkit/core/keras/data_util.py,sha256=jm54o-SlI1DJ-sEvRuX9OyLN68tEt0VxcqrdIjR98Ag,8366
-model_compression_toolkit/core/keras/default_framework_info.py,sha256=YhPSp153YcESp1Ho3GyvoEmxf2CpY9rjTnHAfN7Cpns,6175
-model_compression_toolkit/core/keras/keras_implementation.py,sha256=x5EOYBrg2chC9-OUlrd0laLpnnHCFhYYAFNKRhVh6aQ,28526
-model_compression_toolkit/core/keras/keras_model_validation.py,sha256=dMS9cqaYmliyzVu2-MrKx4AIubqz3HW3RY4if2TV6U8,1581
+model_compression_toolkit/core/keras/default_framework_info.py,sha256=wI-M_MsVIL0qOdBr8F-oyrRz4qdc2o6DDB7lGlizddw,6171
+model_compression_toolkit/core/keras/keras_implementation.py,sha256=5oAc0nc21Uzfo94rKUGGpbYVTmEiKA3YGNhbOwD66ng,28550
 model_compression_toolkit/core/keras/keras_node_prior_info.py,sha256=k9cwu3S-OUGFaOHxH6cyYS2JjxAYHfBddz0laf6Quds,3311
 model_compression_toolkit/core/keras/resource_utilization_data_facade.py,sha256=xxZlHyruhLuP2iEgMrZhq_AyAGORTqzweVLARFfpaRw,5643
 model_compression_toolkit/core/keras/tf_tensor_numpy.py,sha256=jzD8FGEEa8ZD7w8IpTRdp-Udf1MwOTgjg2XTS1Givic,2696
@@ -179,7 +177,7 @@ model_compression_toolkit/core/keras/graph_substitutions/substitutions/batchnorm
 model_compression_toolkit/core/keras/graph_substitutions/substitutions/concat_threshold_update.py,sha256=Hl4LEQ_bw_Vpmf3ZqHujYUqVdvTNsPlEMvr9dZhwg2U,2806
 model_compression_toolkit/core/keras/graph_substitutions/substitutions/conv_funcs_to_layer.py,sha256=vZr8Agj-tFKSX7TM2nZjwbHElJqSIyMAaR7FH-lp4YM,11691
 model_compression_toolkit/core/keras/graph_substitutions/substitutions/dwconv_to_conv.py,sha256=nJO-JUmOK1lLb560KMJgLFwY2IOI2Y3lpzUq7o2f7mQ,5707
-model_compression_toolkit/core/keras/graph_substitutions/substitutions/input_scaling.py,sha256=OwHoCLA-upKUnRpyVWrO_E6QmZcxk6-pOKNpiI7kYzI,6044
+model_compression_toolkit/core/keras/graph_substitutions/substitutions/input_scaling.py,sha256=R3minGoiZqhjpex52LmBPGvALez9KKiuZOQzqOB7OK0,6558
 model_compression_toolkit/core/keras/graph_substitutions/substitutions/linear_collapsing.py,sha256=AvquvVVVT8-ioeVn-gjqysK4L41L3I7TlNOEDfWjViY,8185
 model_compression_toolkit/core/keras/graph_substitutions/substitutions/matmul_substitution.py,sha256=9MZJp4GNTLesWN5uQ5eOQyAHLzLYDAHAjRi-LpNppSc,4257
 model_compression_toolkit/core/keras/graph_substitutions/substitutions/multi_head_attention_decomposition.py,sha256=l9PUREBf4aRwWILiybdteveeUbh7js-i-hLt8Ma0e4c,26771
@@ -201,7 +199,7 @@ model_compression_toolkit/core/keras/mixed_precision/__init__.py,sha256=sw7LOPN1
 model_compression_toolkit/core/keras/mixed_precision/configurable_activation_quantizer.py,sha256=1p2DlMRmgzBAOUP-NeOzldTemjNLQQ3uf1Rov5iY-l8,5430
 model_compression_toolkit/core/keras/mixed_precision/configurable_weights_quantizer.py,sha256=GtW0yars8PzqP9uL_vfXrtqHwKiStmOxPng20rYaIjU,6805
 model_compression_toolkit/core/keras/pruning/__init__.py,sha256=3Lkr37Exk9u8811hw8hVqkGcbTQGcLjd3LLuLC3fa_E,698
-model_compression_toolkit/core/keras/pruning/pruning_keras_implementation.py,sha256=gqlssgSMN3TUzHD_Ple02m6rJHfcW9KpF2ZdTKlH4JM,11312
+model_compression_toolkit/core/keras/pruning/pruning_keras_implementation.py,sha256=tE2-9KqDE2C9Ahs0FzwexCXfO7vviRCLgegn-l5eOTA,11528
 model_compression_toolkit/core/keras/quantization/__init__.py,sha256=mjbqLD-KcG3eNeCYpu1GBS7VclGVOQ63x2p6mAAuba4,698
 model_compression_toolkit/core/keras/quantization/activation_quantization_fn_factory.py,sha256=RtQk5r-bZxUs10AFaJ813_rpkDmOwzWPv6zK6LbX4_8,1876
 model_compression_toolkit/core/keras/quantization/fake_quant_builder.py,sha256=vfKwU0AfRH2KztmMF5bxcaBlGdnTePPGZsUqOHzED-U,6854
@@ -223,7 +221,7 @@ model_compression_toolkit/core/keras/visualization/__init__.py,sha256=mjbqLD-KcG
 model_compression_toolkit/core/pytorch/__init__.py,sha256=Rf1RcYmelmdZmBV5qOKvKWF575ofc06JFQSq83Jz99A,696
 model_compression_toolkit/core/pytorch/constants.py,sha256=Sg0hkUaMe88mI2_pd3KqhVz5ORnA46S1uq9Tj5qhtHc,2828
 model_compression_toolkit/core/pytorch/data_util.py,sha256=YYbT135HhlTt0q6XdD2JX7AS_L92f_uV2rWq2hsJOCA,6325
-model_compression_toolkit/core/pytorch/default_framework_info.py,sha256=pDUE-rwMhm1V1Y19_gwuZDfDCwKAu1ypBvU6XdURVjQ,4308
+model_compression_toolkit/core/pytorch/default_framework_info.py,sha256=HB9JjOBxa0JOLy2bOhDi2Jp2srK8M20g74jOmnUp54I,4305
 model_compression_toolkit/core/pytorch/pytorch_device_config.py,sha256=S25cuw10AW3SEN_fRAGRcG_I3wdvvQx1ehSJzPnn-UI,4404
 model_compression_toolkit/core/pytorch/pytorch_implementation.py,sha256=cUQOBGwtG_DWpkrUEOcYSwXtNSmQgYVBCTxTpFiF4mo,27213
 model_compression_toolkit/core/pytorch/pytorch_node_prior_info.py,sha256=5hsp0nl6TewfrKsT133m9Z7DVpTFFftEv6DeZoryDZw,3009
@@ -272,7 +270,7 @@ model_compression_toolkit/core/pytorch/mixed_precision/__init__.py,sha256=Rf1RcY
 model_compression_toolkit/core/pytorch/mixed_precision/configurable_activation_quantizer.py,sha256=MTH7WsTpP-cTeMwaqrJPnhV_XdFKO6bySNalTONmr0w,4991
 model_compression_toolkit/core/pytorch/mixed_precision/configurable_weights_quantizer.py,sha256=KDnwmbhvhJMfNg1IuTvvzBNEriPQH9bL9dJ5VvWTzpE,6631
 model_compression_toolkit/core/pytorch/pruning/__init__.py,sha256=RAe8mgIr1V8dRIQtLf_dSG5zTUCKuQzxyybYx1dzEAs,697
-model_compression_toolkit/core/pytorch/pruning/pruning_pytorch_implementation.py,sha256=axcG6BKC8gALjjrgOFpiB8b1VbySUyXZIHmzRxQYDoc,13085
+model_compression_toolkit/core/pytorch/pruning/pruning_pytorch_implementation.py,sha256=jO-XbB6-ITnkBWMS4GO_s1-V3fhwfe4c183bMfucAsg,13367
 model_compression_toolkit/core/pytorch/quantization/__init__.py,sha256=Rf1RcYmelmdZmBV5qOKvKWF575ofc06JFQSq83Jz99A,696
 model_compression_toolkit/core/pytorch/quantization/activation_quantization_fn_factory.py,sha256=arslrOgJ1l-fScDlp6jNJ-JukKh0uBLcxAzjpDWRw94,1878
 model_compression_toolkit/core/pytorch/quantization/fake_quant_builder.py,sha256=D8_CEuFqKAhbUgKaRw7Jlxo0zlqgPTMu6CIIIM4LfS0,7045
@@ -368,7 +366,7 @@ model_compression_toolkit/gptq/keras/gptq_keras_implementation.py,sha256=axBwnCS
 model_compression_toolkit/gptq/keras/gptq_loss.py,sha256=2hzWzsbuVd5XcL85NM57YeOyHxRY0qMArKn8NvQ1UWw,7643
 model_compression_toolkit/gptq/keras/gptq_training.py,sha256=_QwytOg1RQSg5Gvme089EME4trdTKGKM3JHIgT-b3n0,22841
 model_compression_toolkit/gptq/keras/graph_info.py,sha256=xpjEqiDo4mqW42QGdmyW31n5eWd6HbYyP6EbarN-A8A,4283
-model_compression_toolkit/gptq/keras/quantization_facade.py,sha256=kA6omL9PoW1hAS2WHN2QoRR1pg2FZQTSyP3qMjDFEJ4,18647
+model_compression_toolkit/gptq/keras/quantization_facade.py,sha256=TBMfJAg_yERdd-OGHFzpV0v435QKM3FvIniwKHjw5i4,18493
 model_compression_toolkit/gptq/keras/quantizer/__init__.py,sha256=-DK1CDXvlsnEbki4lukZLpl6Xrbo91_jcqxXlG5Eg6Q,963
 model_compression_toolkit/gptq/keras/quantizer/base_keras_gptq_quantizer.py,sha256=Rbl9urzkmACvVxICSEyJ02qFOBxWK0UQWtysFJzBVZw,4899
 model_compression_toolkit/gptq/keras/quantizer/quant_utils.py,sha256=Vt7Qb8i4JsE4sFtcjpfM4FTXTtfV1t6SwfoNH8a_Iaw,5055
@@ -403,14 +401,14 @@ model_compression_toolkit/pruning/pytorch/pruning_facade.py,sha256=YxRtJGzD6SjZ4
 model_compression_toolkit/ptq/__init__.py,sha256=Z_hkmTh7aLFei1DJKV0oNVUbrv_Q_0CTw-qD85Xf8UM,904
 model_compression_toolkit/ptq/runner.py,sha256=1tVx3Yj5X4ZjTH0REm6fuAmv4QZ4u_vixLsgjBwBzxc,2326
 model_compression_toolkit/ptq/keras/__init__.py,sha256=cco4TmeIDIh32nj9ZZXVkws4dd9F2UDrmjKzTN8G0V0,697
-model_compression_toolkit/ptq/keras/quantization_facade.py,sha256=_Do07apQ091WCOnVkgJcvnOX812AtXlW0HWx6q3SeRE,11587
+model_compression_toolkit/ptq/keras/quantization_facade.py,sha256=QieIYkoVtxWgHILGYeMHvSeIZ3mdbWdTR76MuUJ08-I,11433
 model_compression_toolkit/ptq/pytorch/__init__.py,sha256=cco4TmeIDIh32nj9ZZXVkws4dd9F2UDrmjKzTN8G0V0,697
 model_compression_toolkit/ptq/pytorch/quantization_facade.py,sha256=RruQVxS4ylBjSH1KMh8ZCV8jk3OvtSrQl24m3Q4xs_8,10065
 model_compression_toolkit/qat/__init__.py,sha256=AaC4KBha4jDW_tyg2SOxZaKh_idIz0gZtDK3_zxs64E,1241
 model_compression_toolkit/qat/common/__init__.py,sha256=6tLZ4R4pYP6QVztLVQC_jik2nES3l4uhML0qUxZrezk,829
 model_compression_toolkit/qat/common/qat_config.py,sha256=QNXj2OcKIJOGvGEGzR2GCifI5Ho7FS7zFc2fkj6PJAc,2750
 model_compression_toolkit/qat/keras/__init__.py,sha256=cco4TmeIDIh32nj9ZZXVkws4dd9F2UDrmjKzTN8G0V0,697
-model_compression_toolkit/qat/keras/quantization_facade.py,sha256=V3-hAO9olSrLCDVezmH1WI8sLrg7q9OrPribL6wn7vI,17429
+model_compression_toolkit/qat/keras/quantization_facade.py,sha256=kwPnkI8nHrUK650-MyJcLweOj7Q7IbFMbmMpP_lRwTg,17275
 model_compression_toolkit/qat/keras/quantizer/__init__.py,sha256=zmYyCa25_KLCSUCGUDRslh3RCIjcRMxc_oXa54Aui-4,996
 model_compression_toolkit/qat/keras/quantizer/base_keras_qat_weight_quantizer.py,sha256=EbIt4lMlh6cU4awFLMBp0IlZ2zUUp-WtnlW5Wn19FDM,1793
 model_compression_toolkit/qat/keras/quantizer/quant_utils.py,sha256=cBULOgWUodcBO1lHevZggdTevuDYI6tQceV86U2x6DA,2543
@@ -473,16 +471,16 @@ model_compression_toolkit/trainable_infrastructure/common/__init__.py,sha256=huH
 model_compression_toolkit/trainable_infrastructure/common/annealing_schedulers.py,sha256=qm2_wa61nga08Jdcl3RkgTsJ0zyHNjZ_A6I2--oVOig,2455
 model_compression_toolkit/trainable_infrastructure/common/base_trainable_quantizer.py,sha256=IF50ASBUvVrOVqlJ1nHNxZxKXSuCanjhUX0YjMB-rRg,7946
 model_compression_toolkit/trainable_infrastructure/common/constants.py,sha256=HN120boJxAnEXNrLSj-o_s-VX4o6C-1ap_KZ4840sd0,875
-model_compression_toolkit/trainable_infrastructure/common/get_quantizer_config.py,sha256=Jxd4IjS_t0FwnA_S_WmZeVbh4VM6Da9ahKGPLp6ZhQo,6983
+model_compression_toolkit/trainable_infrastructure/common/get_quantizer_config.py,sha256=Wv333zLZH0z5N2Uio0beafrUVV-zZcxZvvSKr-fDChY,6839
 model_compression_toolkit/trainable_infrastructure/common/get_quantizers.py,sha256=10edXuhu6F00EcMU7M29AlK7rF_uoLQjMjctrWqK5KU,3346
 model_compression_toolkit/trainable_infrastructure/common/quant_utils.py,sha256=zdiew1jwR7tUKm9XWlHnAPxIZsAdKqbzzC2vH02j5wA,1505
-model_compression_toolkit/trainable_infrastructure/common/trainable_quantizer_config.py,sha256=UXeQpLKYus1BuAc6xKkDMq2iLQUR45s6ATJBa7z4el0,4736
+model_compression_toolkit/trainable_infrastructure/common/trainable_quantizer_config.py,sha256=VLj3YWKJi6MMbEcPq7dDd4DMsXorM6F4Zi7tlFB4YL4,4394
 model_compression_toolkit/trainable_infrastructure/common/training_method.py,sha256=LUoeJkloowhZKuHTiOfzjmSUn2G-4of11-rbnL-h0P4,1194
 model_compression_toolkit/trainable_infrastructure/common/util.py,sha256=oKuWi7E07a8zv5x9auhBugYE2RUQ7ojDh2XCs5koYJY,1090
 model_compression_toolkit/trainable_infrastructure/keras/__init__.py,sha256=huHoBUcKNB6BnY6YaUCcFvdyBtBI172ZoUD8ZYeNc6o,696
 model_compression_toolkit/trainable_infrastructure/keras/annealing_schedulers.py,sha256=sISNVxPsdm-Nd95PhoPSJ-2tFpINGlfrU7ZXaCByI-o,1278
 model_compression_toolkit/trainable_infrastructure/keras/base_keras_quantizer.py,sha256=LBc26z8pkpbcdKMTxpNBg5IyChLreHQ1lRgCVjNE37o,4202
-model_compression_toolkit/trainable_infrastructure/keras/config_serialization.py,sha256=zgGP7G5jXYFe7_hKw9jC2K0bnknKF3LiXpPpBtx-tVM,4304
+model_compression_toolkit/trainable_infrastructure/keras/config_serialization.py,sha256=rsyY7mTev5qRI0ncxuBilgYiRMd5GcuZt6qFyADU4H4,4123
 model_compression_toolkit/trainable_infrastructure/keras/load_model.py,sha256=DJHibcLo-UCuHV6UPLeVd7dKmPfkGXEiLqCCqvQrISM,3769
 model_compression_toolkit/trainable_infrastructure/keras/quantize_wrapper.py,sha256=eVB5FSE3OmTLrhfLUcP2knwN1z2_unQLM-xFEGwdafA,5587
 model_compression_toolkit/trainable_infrastructure/keras/quantizer_utils.py,sha256=r3CaPd4pyM1GDXU2--9NT3wwvl9H6y3QUrVT9spx5es,4189
@@ -532,7 +530,7 @@ model_compression_toolkit/xquant/pytorch/model_analyzer.py,sha256=b93o800yVB3Z-i
 model_compression_toolkit/xquant/pytorch/pytorch_report_utils.py,sha256=Y0oBl8qPFsdNrK49XczwmVacInJcOPHslVnFBs-iTCc,3742
 model_compression_toolkit/xquant/pytorch/similarity_functions.py,sha256=CERxq5K8rqaiE-DlwhZBTUd9x69dtYJlkHOPLB54vm8,2354
 model_compression_toolkit/xquant/pytorch/tensorboard_utils.py,sha256=n0HvWBzkBkUJZlS3WeynhpsRTps2qQkjlq7luliBHNU,9627
-mct_nightly-2.4.0.20250706.701.dist-info/METADATA,sha256=xTdPfpQvxs-8MOaZMaXgKFzbpdzM6z9JhMkBbkKQRzE,25555
-mct_nightly-2.4.0.20250706.701.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mct_nightly-2.4.0.20250706.701.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
-mct_nightly-2.4.0.20250706.701.dist-info/RECORD,,
+mct_nightly-2.4.0.20250708.612.dist-info/METADATA,sha256=DXcxuSC6OdHkGc28g74YdpfdWU7tRPQY-SPtmkfmJvw,25555
+mct_nightly-2.4.0.20250708.612.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mct_nightly-2.4.0.20250708.612.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
+mct_nightly-2.4.0.20250708.612.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.4.0.20250706.000701"
+__version__ = "2.4.0.20250708.000612"

model_compression_toolkit/core/common/collectors/base_collector.py

@@ -13,11 +13,12 @@
 # limitations under the License.
 # ==============================================================================
 
+from abc import ABC, abstractmethod
 import numpy as np
 from model_compression_toolkit.logger import Logger
 
 
-class BaseCollector(object):
+class BaseCollector(ABC):
     """
     Base class for statistics collection object.
     """
@@ -26,6 +27,7 @@ class BaseCollector(object):
         # When manipulation statistics in a granularity they were not collected by, the data is invalid.
         self.is_legal = True
 
+    @abstractmethod
     def scale(self, scale_factor: np.ndarray):
         """
         Scale all statistics in collector by some factor.
@@ -37,6 +39,7 @@ class BaseCollector(object):
         raise NotImplemented(
             f'{self.__class__.__name__} needs to implement scale operation for its state.')  # pragma: no cover
 
+    @abstractmethod
     def shift(self, shift_value: np.ndarray):
         """
         Shift all statistics in collector by some value.

model_compression_toolkit/core/common/collectors/mean_collector.py

@@ -87,10 +87,13 @@ class MeanCollector(BaseCollector):
             x: Tensor that goes through the mean collector and needs to be considered in the mean computation.
         """
         self.i += 1  # Update the iteration index
-        axis = (len(x.shape) - 1) if self.axis == LAST_AXIS else self.axis
-        n = x.shape[axis]
-        transpose_index = [axis, *[i for i in range(len(x.shape)) if i != axis]]
-        mu = np.mean(np.reshape(np.transpose(x, transpose_index), [n, -1]), axis=-1) # mean per channel for a batch
+        if self.axis is None:
+            mu = np.mean(np.reshape(x, [1, -1]), axis=-1)  # mean per channel for a batch
+        else:
+            axis = (len(x.shape) - 1) if self.axis == LAST_AXIS else self.axis
+            n = x.shape[axis]
+            transpose_index = [axis, *[i for i in range(len(x.shape)) if i != axis]]
+            mu = np.mean(np.reshape(np.transpose(x, transpose_index), [n, -1]), axis=-1) # mean per channel for a batch
         self.current_sum += mu # sum of all batches
         self.current_mean = self.current_sum / self.i # mean of all batches
 

model_compression_toolkit/core/common/collectors/min_max_per_channel_collector.py

@@ -130,10 +130,13 @@ class MinMaxPerChannelCollector(BaseCollector):
             x: Tensor that goes through the collector and needs to be considered in the min/max computation.
         """
 
-        axis = (len(x.shape) - 1) if self.axis == LAST_AXIS else self.axis
-        n = x.shape[axis]
-        transpose_index = [axis, *[i for i in range(len(x.shape)) if i != axis]]
-        x_reshape = np.reshape(np.transpose(x, transpose_index), [n, -1])
+        if self.axis is None:
+            x_reshape = np.reshape(x, [1, -1])
+        else:
+            axis = (len(x.shape) - 1) if self.axis == LAST_AXIS else self.axis
+            n = x.shape[axis]
+            transpose_index = [axis, *[i for i in range(len(x.shape)) if i != axis]]
+            x_reshape = np.reshape(np.transpose(x, transpose_index), [n, -1])
         if self.state is None:
             x_max = np.max(x_reshape, axis=-1)
             x_min = np.min(x_reshape, axis=-1)

model_compression_toolkit/core/common/model_collector.py

@@ -57,19 +57,21 @@ def create_stats_collector_for_node(node: common.BaseNode,
 
 
 def create_tensor2node(graph: common.Graph,
-                       node: common.BaseNode):
+                       node: common.BaseNode,
+                       next_node_output_channel_axis: int):
     """
     Force statistic collector creation and assignment for a node.
     Args:
         graph: Graph of the node (for retrieving the current tensor).
         node: Node to create a tensor for.
+        next_node_output_channel_axis: channel output axis of next node.
 
     """
     current_sc = graph.get_out_stats_collector(node)
     is_list_nostat_collectors = isinstance(current_sc, list) and len(
         [sc for sc in current_sc if not isinstance(sc, common.NoStatsCollector)]) == 0
     if isinstance(current_sc, common.NoStatsCollector) or current_sc is None or is_list_nostat_collectors:
-        stats_collector = common.StatsCollector(node.out_channel_axis)
+        stats_collector = common.StatsCollector(next_node_output_channel_axis if node.out_channel_axis is None else node.out_channel_axis)
         graph.set_out_stats_collector_to_node(node, stats_collector)
 
 
@@ -157,6 +159,17 @@ class ModelCollector:
         for n in graph.get_topo_sorted_nodes():
             quant_node_in_fln = n.is_fln_quantization() and graph.fusing_info.is_quantized_node_in_fln(n)
             sc = create_stats_collector_for_node(n, quant_node_in_fln=quant_node_in_fln)  # Get static collector for the node
+            if isinstance(sc, common.StatsCollector) and (sc.mc.axis is None or sc.mpcc.axis is None):
+                # Missing output channel axis info, so try to extract it from previous and next nodes output channel axis.
+                possible_output_channel_axis_set = {nn.out_channel_axis for nn in graph.get_next_nodes(n) + graph.get_prev_nodes(n)}
+                # Filter out None values.
+                possible_output_channel_axis_list = list(filter(lambda x: x is not None, possible_output_channel_axis_set))
+                if len(possible_output_channel_axis_list) > 0:
+                    if len(possible_output_channel_axis_list) > 1:
+                        Logger.warning(f'Ambiguous input channel data from next nodes for {n.name}.')
+                    sc.mc.axis = possible_output_channel_axis_list[0]
+                    sc.mpcc.axis = possible_output_channel_axis_list[0]
+
             # If we use bias correction, and the node has kernel weights to quantize, we need to make sure
             # its previous nodes' tensors are consistent with this node.
             if qc.weights_bias_correction and n.kernel_attr is not None and n.is_weights_quantization_enabled(
@@ -164,7 +177,8 @@ class ModelCollector:
                 for ie in graph.incoming_edges(n):
                     input_node = ie.source_node
                     create_tensor2node(graph,
-                                       input_node)
+                                       input_node,
+                                       n.out_channel_axis)
             if sc is not None:
                 graph.set_out_stats_collector_to_node(n, sc)
 

model_compression_toolkit/core/common/pruning/memory_calculator.py

@@ -303,7 +303,7 @@ class MemoryCalculator:
             num_oc = np.sum(output_mask)
         else:
             # Get the node channel axis from framework info
-            channel_axis = node.out_channel_axis
+            channel_axis = self.fw_impl.default_output_channel_axis if node.out_channel_axis is None else node.out_channel_axis
             if channel_axis is None:
                 Logger.critical(f"The channel axis is undefined. Please ensure the channel axis is explicitly defined for node {node.type} in the framework info.")
 

model_compression_toolkit/core/common/quantization/node_quantization_config.py

@@ -18,7 +18,6 @@ from enum import Enum, auto
 from model_compression_toolkit.core.common.framework_info import ChannelAxisMapping
 from model_compression_toolkit.logger import Logger
 
-from model_compression_toolkit.core.common.quantization.quantization_config import QuantizationConfig
 from model_compression_toolkit.target_platform_capabilities.constants import POSITIONAL_ATTR
 from model_compression_toolkit.target_platform_capabilities.schema.mct_current_schema import \
     AttributeQuantizationConfig, OpQuantizationConfig
@@ -41,6 +40,7 @@ class ActivationQuantizationMode(Enum):
     NO_QUANT = auto()
     FLN_NO_QUANT = auto()
 
+
 class BaseNodeQuantizationConfig(object):
     """
     Base class for node quantization configuration
@@ -59,12 +59,11 @@ class BaseNodeQuantizationConfig(object):
             kwargs: A dictionary with additional key arguments.
 
         """
-
         if hasattr(self, config_parameter_name):
             setattr(self, config_parameter_name, config_parameter_value)
         else:
-            Logger.warning(f"Parameter {config_parameter_name} could not be found in the node quantization config and "
-                           f"was not updated!")
+            raise AttributeError(
+                f"Parameter {config_parameter_name} could not be found in the node quantization config.")
 
     def __repr__(self) -> str:
         """
@@ -97,36 +96,11 @@ class NodeActivationQuantizationConfig(BaseNodeQuantizationConfig):
         self.signedness = op_cfg.signedness
 
         self.activation_quantization_params = {}
-        # TODO irena: computed by compute_activation_bias_correction. shouldnt really be here
+        # TODO: computed by compute_activation_bias_correction. Probably shouldnt be here.
         self.activation_bias_correction_term = None
-
-        # TODO irena remove along with set_qc. Keeping for eq and hash to work without set_qc being called
-        self.activation_error_method = None
-        self.relu_bound_to_power_of_2 = None
-        self.activation_channel_equalization = None
-        self.input_scaling = None
-        self.min_threshold = None
-        self.l_p_value = None
-        self.shift_negative_activation_correction = None
+        # Z-threshold is a global param from QuantizationConfig, however it can be overridden per node by NetworkEditor.
+        # Since activation qparams are re-computed in several places, it's easier to keep it here and update it once.
         self.z_threshold = None
-        self.shift_negative_ratio = None
-        self.shift_negative_threshold_recalculation = None
-        self.concat_threshold_update = None
-
-    def set_qc(self, qc: QuantizationConfig):
-        """ TODO irena: temporary keep all the attributes as before not to break all code at once.
-             Eventually all of them should be removed from here. """
-        self.activation_error_method = qc.activation_error_method
-        self.relu_bound_to_power_of_2 = qc.relu_bound_to_power_of_2
-        self.activation_channel_equalization = qc.activation_channel_equalization
-        self.input_scaling = qc.input_scaling
-        self.min_threshold = qc.min_threshold
-        self.l_p_value = qc.l_p_value
-        self.shift_negative_activation_correction = qc.shift_negative_activation_correction
-        self.z_threshold = qc.z_threshold
-        self.shift_negative_ratio = qc.shift_negative_ratio
-        self.shift_negative_threshold_recalculation = qc.shift_negative_threshold_recalculation
-        self.concat_threshold_update = qc.concat_threshold_update
 
     @property
     def enable_activation_quantization(self):
@@ -148,7 +122,7 @@ class NodeActivationQuantizationConfig(BaseNodeQuantizationConfig):
             activation_params: Dictionary that contains weight quantization params.
 
         """
-        assert self.quant_mode == ActivationQuantizationMode.QUANT
+        assert self.quant_mode == ActivationQuantizationMode.QUANT or self.quant_mode == ActivationQuantizationMode.FLN_QUANT
         for param_name, param_value in activation_params.items():
             self.activation_quantization_params[param_name] = param_value
 
@@ -165,32 +139,16 @@ class NodeActivationQuantizationConfig(BaseNodeQuantizationConfig):
         if not isinstance(other, NodeActivationQuantizationConfig):
             return False  # pragma: no cover
 
-        return self.activation_error_method == other.activation_error_method and \
-               self.activation_quantization_method == other.activation_quantization_method and \
+        return self.activation_quantization_method == other.activation_quantization_method and \
                self.activation_n_bits == other.activation_n_bits and \
                self.quant_mode == other.quant_mode and \
-               self.activation_channel_equalization == other.activation_channel_equalization and \
-               self.input_scaling == other.input_scaling and \
-               self.min_threshold == other.min_threshold and \
-               self.l_p_value == other.l_p_value and \
-               self.shift_negative_activation_correction == other.shift_negative_activation_correction and \
-               self.z_threshold == other.z_threshold and \
-               self.shift_negative_ratio == other.shift_negative_ratio and \
-               self.shift_negative_threshold_recalculation == other.shift_negative_threshold_recalculation
+               self.signedness == other.signedness
 
     def __hash__(self):
-        return hash((self.activation_error_method,
-                     self.activation_quantization_method,
+        return hash((self.activation_quantization_method,
                      self.activation_n_bits,
                      self.quant_mode,
-                     self.activation_channel_equalization,
-                     self.input_scaling,
-                     self.min_threshold,
-                     self.l_p_value,
-                     self.shift_negative_activation_correction,
-                     self.z_threshold,
-                     self.shift_negative_ratio,
-                     self.shift_negative_threshold_recalculation))
+                     self.signedness))
 
 
 class WeightsAttrQuantizationConfig:
@@ -211,16 +169,8 @@ class WeightsAttrQuantizationConfig:
         self.weights_n_bits = weights_attr_cfg.weights_n_bits
         self.weights_per_channel_threshold = weights_attr_cfg.weights_per_channel_threshold
         self.enable_weights_quantization = weights_attr_cfg.enable_weights_quantization
-        self.weights_quantization_params = {}
 
-        # TODO irena remove along with set_qc. Keeping for eq and hash to work without set_qc being called
-        self.weights_error_method = None
-        self.l_p_value = None
-
-    def set_qc(self, qc: QuantizationConfig):
-        # TODO irena: temporary keep the fields to not break everything at once.
-        self.weights_error_method = qc.weights_error_method
-        self.l_p_value = qc.l_p_value
+        self.weights_quantization_params = {}
 
     def set_weights_quantization_param(self,
                                        weights_params: dict):
@@ -252,18 +202,14 @@ class WeightsAttrQuantizationConfig:
                self.weights_quantization_method == other.weights_quantization_method and \
                self.weights_n_bits == other.weights_n_bits and \
                self.weights_per_channel_threshold == other.weights_per_channel_threshold and \
-               self.enable_weights_quantization == other.enable_weights_quantization and \
-               self.weights_error_method == other.weights_error_method and \
-               self.l_p_value == other.l_p_value
+               self.enable_weights_quantization == other.enable_weights_quantization
 
     def __hash__(self):
         return hash((self.weights_channels_axis,
-                     self.weights_error_method,
                      self.weights_quantization_method,
                      self.weights_n_bits,
                      self.weights_per_channel_threshold,
-                     self.enable_weights_quantization,
-                     self.l_p_value))
+                     self.enable_weights_quantization))
 
 
 class NodeWeightsQuantizationConfig(BaseNodeQuantizationConfig):
@@ -330,16 +276,14 @@ class NodeWeightsQuantizationConfig(BaseNodeQuantizationConfig):
 
                 self.attributes_config_mapping[attr] = WeightsAttrQuantizationConfig(weights_attr_cfg=attr_cfg,
                                                                                      weights_channels_axis=weights_channels_axis)
-        # TODO irena remove along with set_qc. Keeping for eq and hash to work without set_qc being called
-        self.min_threshold = None
+        # TODO this is set by batch norm reconstruction substitution when folded batch norms are added back, to mark
+        #  the nodes that the correction should be applied to (for some nodes it gets disabled) and BNs removed.
+        #  The actual correction is only computed when it's applied in ptq, so it seems that both substitutions could
+        #  be unified, and no info need to pass between.
         self.weights_second_moment_correction = None
-        self.weights_bias_correction = None
-
-    def set_qc(self, qc: QuantizationConfig):
-        # TODO irena: temporary keep the fields to not break everything at once.
-        self.min_threshold = qc.min_threshold
-        self.weights_second_moment_correction = qc.weights_second_moment_correction
-        self.weights_bias_correction = qc.weights_bias_correction
+        # TODO: computed corrected bias is injected to the node config. Probably shouldn't be here. Also it can be
+        #  computed on the final config, instead of all candidates and then there is no need to save it at all.
+        self.bias_corrected = None
 
     def get_attr_config(self, attr_name: 'WeightAttrT') -> WeightsAttrQuantizationConfig:
         """
@@ -476,8 +420,8 @@ class NodeWeightsQuantizationConfig(BaseNodeQuantizationConfig):
                 if hasattr(attr_cfg, config_parameter_name):
                     setattr(attr_cfg, config_parameter_name, config_parameter_value)
                 else:
-                    Logger.warning(f"Parameter {config_parameter_name} could not be found in the node quantization config of "
-                                   f"weights attribute {attr_name} and was not updated!")
+                    raise AttributeError(f"Parameter {config_parameter_name} could not be found in the node quantization config of "
+                                         f"weights attribute {attr_name}.")
             else:  # pragma: no cover
                 Logger.critical(f"Weights attribute {attr_name} could not be found to set parameter {config_parameter_name}.")
 
@@ -494,10 +438,7 @@ class NodeWeightsQuantizationConfig(BaseNodeQuantizationConfig):
         if not isinstance(other, NodeWeightsQuantizationConfig):
             return False  # pragma: no cover
 
-        return self.min_threshold == other.min_threshold and \
-            self.simd_size == other.simd_size and \
-            self.weights_second_moment_correction == other.weights_second_moment_correction and \
-            self.weights_bias_correction == other.weights_bias_correction and \
+        return self.simd_size == other.simd_size and \
             self.attributes_config_mapping.keys() == other.attributes_config_mapping.keys() and \
             all([self.attributes_config_mapping[k] == other.attributes_config_mapping[k]
                  for k in self.attributes_config_mapping.keys()]) and \
@@ -506,9 +447,6 @@ class NodeWeightsQuantizationConfig(BaseNodeQuantizationConfig):
                  for k in self.pos_attributes_config_mapping.keys()])
 
     def __hash__(self):
-        return hash((self.min_threshold,
-                     self.simd_size,
-                     self.weights_second_moment_correction,
-                     self.weights_bias_correction,
+        return hash((self.simd_size,
                      frozenset(self.attributes_config_mapping),
                      frozenset(self.pos_attributes_config_mapping)))

model_compression_toolkit/core/common/quantization/quantization_config.py

@@ -90,7 +90,6 @@ class QuantizationConfig:
     shift_negative_activation_correction: bool = True
     activation_channel_equalization: bool = False
     z_threshold: float = math.inf
-    min_threshold: float = MIN_THRESHOLD
     l_p_value: int = 2
     linear_collapsing: bool = True
     residual_collapsing: bool = True