mct-nightly 1.9.0.20230814.post352__py3-none-any.whl → 1.9.0.20230816.post359__py3-none-any.whl

{mct_nightly-1.9.0.20230814.post352.dist-info → mct_nightly-1.9.0.20230816.post359.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mct-nightly
-Version: 1.9.0.20230814.post352
+Version: 1.9.0.20230816.post359
 Summary: A Model Compression Toolkit for neural networks
 Home-page: UNKNOWN
 License: UNKNOWN
@@ -91,9 +91,9 @@ MCT supports compressing models built with the TensorFlow or PyTorch frameworks,
 
 | TensorFlow Version                                                                                   | PyTorch Version                                                                                          |
 |------------------------------------------------------------------------------------------------------|----------------------------------------------------------------------------------------------------------|
-| ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf212.yml/badge.svg) | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_13.yml/badge.svg) |
-| ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf211.yml/badge.svg) | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_12.yml/badge.svg) |
-| ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf210.yml/badge.svg) | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_11.yml/badge.svg) |
+| ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf213.yml/badge.svg) | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_13.yml/badge.svg) |
+| ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf212.yml/badge.svg) | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_12.yml/badge.svg) |
+| ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_tf211.yml/badge.svg) | ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_torch1_11.yml/badge.svg) |
 
 
 ## Supported Features

{mct_nightly-1.9.0.20230814.post352.dist-info → mct_nightly-1.9.0.20230816.post359.dist-info}/RECORD

@@ -129,19 +129,19 @@ model_compression_toolkit/core/common/visualization/tensorboard_writer.py,sha256
 model_compression_toolkit/core/keras/__init__.py,sha256=mjbqLD-KcG3eNeCYpu1GBS7VclGVOQ63x2p6mAAuba4,698
 model_compression_toolkit/core/keras/constants.py,sha256=nbty5PUcLhC9aTuAemXTEmKcPF--7a5adO1-nQsgB9A,2664
 model_compression_toolkit/core/keras/custom_layer_validation.py,sha256=f-b14wuiIgitBe7d0MmofYhDCTO3IhwJgwrh-Hq_t_U,1192
-model_compression_toolkit/core/keras/default_framework_info.py,sha256=QsaZRdWJqz4ev3ZQZCkUfg9Imu8iyEL_6NRFCSVHQkE,4999
-model_compression_toolkit/core/keras/keras_implementation.py,sha256=rYiN1MaKkz183wvvmUrbQL2crROBBQSA70acTmC3mn8,27448
+model_compression_toolkit/core/keras/default_framework_info.py,sha256=HLLO28tsbb9tHYQ05grUr3oJTRKdS520QnyGUYgzRK0,4994
+model_compression_toolkit/core/keras/keras_implementation.py,sha256=9qD5s9XmxGzTd_6MG4tWj_n-nibHOf7XBr8WYzwTY_I,27439
 model_compression_toolkit/core/keras/keras_model_validation.py,sha256=1wNV2clFdC9BzIELRLSO2uKf0xqjLqlkTJudwtCeaJk,1722
-model_compression_toolkit/core/keras/keras_node_prior_info.py,sha256=KuC6yWUUASoOCxK5ZybI7dVydveLCovUTfPmIig9CTg,3941
+model_compression_toolkit/core/keras/keras_node_prior_info.py,sha256=f6o5Fmpw0aDrO704_A-SqBrKSO1iNEOyofP9pm3g8yg,3936
 model_compression_toolkit/core/keras/kpi_data_facade.py,sha256=rArrfMtxWGR1P4nhKKxqh6fo7pauRDzkRsZIh_SXxO4,8502
 model_compression_toolkit/core/keras/tf_tensor_numpy.py,sha256=BauH-Ssoiuv5wu81fk1bm9HO_0yR0oTwKyLkXgE3qSE,2022
 model_compression_toolkit/core/keras/back2framework/__init__.py,sha256=rhIiXg_nBgUZ-baE3M6SzCuQbcnq4iebY1jtJBvKHOM,808
 model_compression_toolkit/core/keras/back2framework/factory_model_builder.py,sha256=GSh1Piz5qpA7IlvHTMqUvPn7WBDa0IHEDZdd_TzY9XA,2226
 model_compression_toolkit/core/keras/back2framework/float_model_builder.py,sha256=9SFHhX-JnkB8PvYIIHRYlReBDI_RkZY9LditzW_ElLk,2444
 model_compression_toolkit/core/keras/back2framework/instance_builder.py,sha256=TOn_MEp0UCDshy6mKiHeHuKGCWW_Xz81V0o_POVmgLc,4508
-model_compression_toolkit/core/keras/back2framework/keras_model_builder.py,sha256=0FB4TdbGp34ejBhOfqZmi_249YokaypLSpBbAi8WIdk,16109
-model_compression_toolkit/core/keras/back2framework/mixed_precision_model_builder.py,sha256=5getD3pU1ma2MqAKtHlNud_uCCW1UC42zJLGnXVqFIE,14442
-model_compression_toolkit/core/keras/back2framework/model_gradients.py,sha256=d7lmx6da978Oe62mQwa8aloQayd4Nygrs0sxBE0qIVQ,15431
+model_compression_toolkit/core/keras/back2framework/keras_model_builder.py,sha256=l2l7JoyZLfKt10xuX4s8--SWpxlhMV1ZHOme2rmnaww,15916
+model_compression_toolkit/core/keras/back2framework/mixed_precision_model_builder.py,sha256=DtH6c_eKYw6rq8oQcX-R6Ddbz3uX8v6jmiHeowerL9w,14616
+model_compression_toolkit/core/keras/back2framework/model_gradients.py,sha256=IT5LJoSFqIqGFVtQpu-LUQ10TJyaZiRPnImqwQCI0iA,15428
 model_compression_toolkit/core/keras/back2framework/quantized_model_builder.py,sha256=5wFb4nx_F0Wu4c8pLf6n6OzxOHtpOJ6_3mQsNSXIudU,2481
 model_compression_toolkit/core/keras/graph_substitutions/__init__.py,sha256=mjbqLD-KcG3eNeCYpu1GBS7VclGVOQ63x2p6mAAuba4,698
 model_compression_toolkit/core/keras/graph_substitutions/substitutions/__init__.py,sha256=mjbqLD-KcG3eNeCYpu1GBS7VclGVOQ63x2p6mAAuba4,698
@@ -151,13 +151,13 @@ model_compression_toolkit/core/keras/graph_substitutions/substitutions/batchnorm
 model_compression_toolkit/core/keras/graph_substitutions/substitutions/batchnorm_refusing.py,sha256=5df_xGfXkqNub4xVRnCWQvSohWqdv12axjJ6edVU2H0,2478
 model_compression_toolkit/core/keras/graph_substitutions/substitutions/input_scaling.py,sha256=Yj59BMBrITJnXJHH-7de91LJwH_1l1WhY1udSQjdoi4,5598
 model_compression_toolkit/core/keras/graph_substitutions/substitutions/linear_collapsing.py,sha256=Yl72Io4-etnsOXfMKAZmC2lDzmskxZu5gey7IBcUukU,5925
-model_compression_toolkit/core/keras/graph_substitutions/substitutions/multi_head_attention_decomposition.py,sha256=VpJwWWW6zPkb7_giaRJ801qJ2DIKkQiBYN0d7NvDWIQ,26778
+model_compression_toolkit/core/keras/graph_substitutions/substitutions/multi_head_attention_decomposition.py,sha256=aAG5wrcnnydn1pPYqvH56LWsQXjSODbsoNbX_jtQGP4,26759
 model_compression_toolkit/core/keras/graph_substitutions/substitutions/relu_bound_to_power_of_2.py,sha256=IdKOg6AWZWMcmDbOuNdxetS5_zTarXIIffdYL7JTdvk,3872
 model_compression_toolkit/core/keras/graph_substitutions/substitutions/remove_relu_upper_bound.py,sha256=cJQTDzTDQKAJ7EQ20tfsmReGA_OoTIN793MwVe1Ok8g,2387
 model_compression_toolkit/core/keras/graph_substitutions/substitutions/residual_collapsing.py,sha256=6PnPIC5ax7uTzcoslW7ropIu7vVmo70AD4QYcYnQV20,3176
 model_compression_toolkit/core/keras/graph_substitutions/substitutions/scale_equalization.py,sha256=ryes9y1ie-vjBGso2TeO4EXxVk69Ew3iSAhshPz1Ou4,5542
 model_compression_toolkit/core/keras/graph_substitutions/substitutions/separableconv_decomposition.py,sha256=OqtLx3yeUXlE1DfA1dA4oWk0yVSMPSgvYb0dQ8Ytb5E,7730
-model_compression_toolkit/core/keras/graph_substitutions/substitutions/shift_negative_activation.py,sha256=g675BwcNWr6SqEW9J-Ys2i8T4XaxzXpjYNUlR0gtHfM,10781
+model_compression_toolkit/core/keras/graph_substitutions/substitutions/shift_negative_activation.py,sha256=6vEakr0jWrccU7dfubRCiNg6TFe6whte_pbTiXMJIvc,11045
 model_compression_toolkit/core/keras/graph_substitutions/substitutions/softmax_shift.py,sha256=Qk5seDALj_th9dHJehY7ynZjvFjVfCv_mJ1enA5hX0c,1623
 model_compression_toolkit/core/keras/graph_substitutions/substitutions/virtual_activation_weights_composition.py,sha256=wH9ocMLL725-uUPU-zCxdd8NwT5nyd0ZShmI7iuTwF8,1462
 model_compression_toolkit/core/keras/graph_substitutions/substitutions/weights_activation_split.py,sha256=rjIheZW7LbSPv9bzMSmC8wl6UUxaTkd4J2IHinObT-Y,1814
@@ -169,9 +169,9 @@ model_compression_toolkit/core/keras/quantizer/base_quantizer.py,sha256=eMRjAUU1
 model_compression_toolkit/core/keras/quantizer/fake_quant_builder.py,sha256=Oi64CD83OopPoQNAarl2MJRbCKujU2W8Wdrs9KOPNWk,6151
 model_compression_toolkit/core/keras/quantizer/lut_fake_quant.py,sha256=Up3-sbuAcaJ6kfe7Sz3XN6iiJ9hlxzOMncLCFEXJFjk,4475
 model_compression_toolkit/core/keras/reader/__init__.py,sha256=mjbqLD-KcG3eNeCYpu1GBS7VclGVOQ63x2p6mAAuba4,698
-model_compression_toolkit/core/keras/reader/common.py,sha256=ScJaCiM8FuisN3sLbVsMQiIzkFlzr-gYEeL8M5lkoew,2627
-model_compression_toolkit/core/keras/reader/connectivity_handler.py,sha256=83Abmg-AILnN6NHJexI226ZaHEblLOX3H9sH6vXFQhA,11418
-model_compression_toolkit/core/keras/reader/node_builder.py,sha256=3wHqvJ8AA7ZEtYWRXYIlIDOLwdixha3Nt1cyCgdpGpk,6408
+model_compression_toolkit/core/keras/reader/common.py,sha256=z0PMoP_HjndN3upYTEQS6yxSGqQsM74V3uRHt_Hx3jw,2490
+model_compression_toolkit/core/keras/reader/connectivity_handler.py,sha256=AgF6qXZOJMeXvc-pBnGY23BJz7wPBx2aTYxHiO8efec,11303
+model_compression_toolkit/core/keras/reader/node_builder.py,sha256=eP9FOcMePxXOHiDr-BGn3YKVhWc11HmN8UIcQyMgJAg,6368
 model_compression_toolkit/core/keras/reader/reader.py,sha256=wS9UQ2wJKnkZYe9JHwQp7ygDr6CRlzrxmIyLDv1Qz6U,8109
 model_compression_toolkit/core/keras/reader/nested_model/__init__.py,sha256=mjbqLD-KcG3eNeCYpu1GBS7VclGVOQ63x2p6mAAuba4,698
 model_compression_toolkit/core/keras/reader/nested_model/edges_merger.py,sha256=K6KAH9o8KSG6baLmhKoCrYK-i-wb6gRKiZmoijFqEYA,7906
@@ -235,7 +235,7 @@ model_compression_toolkit/exporter/model_exporter/fw_agonstic/exporter.py,sha256
 model_compression_toolkit/exporter/model_exporter/keras/__init__.py,sha256=uZ2RigbY9O2PJ0Il8wPpS_s7frgg9WUGd_SHeKGyl1A,699
 model_compression_toolkit/exporter/model_exporter/keras/base_keras_exporter.py,sha256=93UkXZEm6sTCATUwLu4exVyIFAFBzbA4Yg_Tr89Rb9U,1495
 model_compression_toolkit/exporter/model_exporter/keras/export_serialization_format.py,sha256=coIfpkMTfPImzPb1LpK_HYzyZ5h5cGBbPoJGHHcoIC8,966
-model_compression_toolkit/exporter/model_exporter/keras/fakely_quant_keras_exporter.py,sha256=8gGZVN45rOSdt9kKRmx1w0aS2t3HqvxKB1v1z1ow7fg,11100
+model_compression_toolkit/exporter/model_exporter/keras/fakely_quant_keras_exporter.py,sha256=o8IyEgHdrjk2OvWBrxl11oGH0lVvqX8aynVUYCAD4mk,11285
 model_compression_toolkit/exporter/model_exporter/keras/fakely_quant_tflite_exporter.py,sha256=sgTlqNjHSf0vVzQpq2811WL6k7SIo_QZaV7TxUtIC8I,3043
 model_compression_toolkit/exporter/model_exporter/keras/int8_tflite_exporter.py,sha256=nGtpDTeH5Tdp7sjyuXsy_9TPpijDYp4nkz366DUUJ0Q,8048
 model_compression_toolkit/exporter/model_exporter/keras/keras_export_facade.py,sha256=mGaIaMGQ3qVWMEbosyL0uYc-3KsjXumexoaTbeue1X4,5982
@@ -247,7 +247,7 @@ model_compression_toolkit/exporter/model_exporter/pytorch/fakely_quant_torchscri
 model_compression_toolkit/exporter/model_exporter/pytorch/pytorch_export_facade.py,sha256=rzUd377eyHHo0dD6z0efTTR-8Bo6Ebqo0X8ECL-Q7Gw,5866
 model_compression_toolkit/exporter/model_wrapper/__init__.py,sha256=7CF2zvpTrIEm8qnbuHnLZyTZkwBBxV24V8QA0oxGbh0,1187
 model_compression_toolkit/exporter/model_wrapper/keras/__init__.py,sha256=cco4TmeIDIh32nj9ZZXVkws4dd9F2UDrmjKzTN8G0V0,697
-model_compression_toolkit/exporter/model_wrapper/keras/validate_layer.py,sha256=XMXcszmMSEgeIc1tCNZqWjsyFt9ZmcgfWZQXDqnDMjM,3509
+model_compression_toolkit/exporter/model_wrapper/keras/validate_layer.py,sha256=LzwUapHiTuXM8b7qFkoK5XvkNNJynhdw4L1A6XTKTJs,3767
 model_compression_toolkit/exporter/model_wrapper/keras/builder/__init__.py,sha256=cco4TmeIDIh32nj9ZZXVkws4dd9F2UDrmjKzTN8G0V0,697
 model_compression_toolkit/exporter/model_wrapper/keras/builder/fully_quantized_model_builder.py,sha256=P_701NG1OYwyxgbtPSjb-OYseDeiHjye8m9wjpAl6es,4253
 model_compression_toolkit/exporter/model_wrapper/keras/builder/node_to_quantizer.py,sha256=VFX8rvlaGTK-k-3g0ayAL_qJJR_6wtS54tn9RzxKev4,8755
@@ -256,7 +256,7 @@ model_compression_toolkit/exporter/model_wrapper/pytorch/__init__.py,sha256=Rf1R
 model_compression_toolkit/exporter/model_wrapper/pytorch/validate_layer.py,sha256=gvX5ILs5vjQ_F_dq5KaFs0GOQEq9gYXO5a6YZlYY8h4,3449
 model_compression_toolkit/exporter/model_wrapper/pytorch/builder/__init__.py,sha256=cco4TmeIDIh32nj9ZZXVkws4dd9F2UDrmjKzTN8G0V0,697
 model_compression_toolkit/exporter/model_wrapper/pytorch/builder/fully_quantized_model_builder.py,sha256=SJ5fetbUMkmB0tkHkmVhMrLksh7eqMQJLFuMD08ZKWM,3921
-model_compression_toolkit/exporter/model_wrapper/pytorch/builder/node_to_quantizer.py,sha256=66QXUQI1yvov9QXCW1s8LLN9N7gFwvVrPJDfzHayWZM,8635
+model_compression_toolkit/exporter/model_wrapper/pytorch/builder/node_to_quantizer.py,sha256=62dI97t4N7TO7xDYnn3992B17XYKryyO9ZhxV1vdThY,8626
 model_compression_toolkit/exporter/model_wrapper/pytorch/builder/node_to_quantizers.py,sha256=hinP-wtyxZyoW860GdJAk6M3iPjmwwPXQTUxd56yhq8,2086
 model_compression_toolkit/gptq/__init__.py,sha256=2xos6AJziEy-eK91XtIJlunf8LhK4OayU7d6CQvXWsw,1276
 model_compression_toolkit/gptq/runner.py,sha256=vWd7cWKgTGc9oPcTtwTQZoI3MArCx19Y61uteLFCxVo,5534
@@ -269,7 +269,7 @@ model_compression_toolkit/gptq/common/gptq_training.py,sha256=U24sNWiVzXEfnk4ePO
 model_compression_toolkit/gptq/keras/__init__.py,sha256=cco4TmeIDIh32nj9ZZXVkws4dd9F2UDrmjKzTN8G0V0,697
 model_compression_toolkit/gptq/keras/gptq_keras_implementation.py,sha256=axBwnCSjq5xk-xGymOwSOqjp39It-CVtGcCTRTf0E_4,1248
 model_compression_toolkit/gptq/keras/gptq_loss.py,sha256=rbRkF15MYd6nq4G49kcjb_dPTa-XNq9cTkrb93mXawo,6241
-model_compression_toolkit/gptq/keras/gptq_training.py,sha256=49HM4zj5BMlTzEtFGLdEqk0upvuKTYlvCMx9okzNyNk,17319
+model_compression_toolkit/gptq/keras/gptq_training.py,sha256=gliTu3hoL3bOMytgeaUfbcJ0bz7-bih-Lm3CkMvQiOM,17307
 model_compression_toolkit/gptq/keras/graph_info.py,sha256=B9wMdlnUNHaFog6UsE3bEZrcB6j1ySNYaU4o_kI91H0,4564
 model_compression_toolkit/gptq/keras/quantization_facade.py,sha256=siVgRlNhWwmiA-9FN7MIs3XSEp0NjgCPnXvBkFu5GLQ,14835
 model_compression_toolkit/gptq/keras/quantizer/__init__.py,sha256=-DK1CDXvlsnEbki4lukZLpl6Xrbo91_jcqxXlG5Eg6Q,963
@@ -328,8 +328,8 @@ model_compression_toolkit/qat/pytorch/quantizer/base_pytorch_qat_quantizer.py,sh
 model_compression_toolkit/qat/pytorch/quantizer/quantization_builder.py,sha256=GOYRDXvQSGe_iUFVmvDy5BqC952hu_-rQO06n8QCyw0,5491
 model_compression_toolkit/qat/pytorch/quantizer/quantizer_utils.py,sha256=5XswoF-5aaFangkHssWKAQTsk6lf_zzndzfCsBWBVMs,5004
 model_compression_toolkit/qat/pytorch/quantizer/ste_rounding/__init__.py,sha256=Rf1RcYmelmdZmBV5qOKvKWF575ofc06JFQSq83Jz99A,696
-model_compression_toolkit/qat/pytorch/quantizer/ste_rounding/symmetric_ste.py,sha256=OeVcB9w5kIvP-KJiI-cBJNnLt-rUCEqEeVXJU5GInkQ,9621
-model_compression_toolkit/qat/pytorch/quantizer/ste_rounding/uniform_ste.py,sha256=bA2XfqGD3TPz_5XhaKU0dzxfNI_v0c8_0UywJGjKumM,8643
+model_compression_toolkit/qat/pytorch/quantizer/ste_rounding/symmetric_ste.py,sha256=4xmLmg7yN2A7iKnifwkWddgJTWMUiIjFilIuorJeK1A,9657
+model_compression_toolkit/qat/pytorch/quantizer/ste_rounding/uniform_ste.py,sha256=0zFu84d49PF5GtgFbgu1ISxhqOqBG0_U9CdOhBiKnBk,8712
 model_compression_toolkit/target_platform_capabilities/__init__.py,sha256=cco4TmeIDIh32nj9ZZXVkws4dd9F2UDrmjKzTN8G0V0,697
 model_compression_toolkit/target_platform_capabilities/constants.py,sha256=mYUESjXeN7EPomZKgqHZE031KNOO4wmFdRB8gA4m50U,920
 model_compression_toolkit/target_platform_capabilities/immutable.py,sha256=rSPd3Xpx4Rzt6-EwDA9tXvKygrrt4xvmq01JVCXY0hQ,1723
@@ -355,56 +355,56 @@ model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/ta
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/latest/__init__.py,sha256=4Yr2tWrFAVDh_41o5TF0VUusMMyg0FIFaPWpW1tLQ5w,1514
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v1/__init__.py,sha256=t4JKsPcor-7KSCKzIwuaBv0NLNwfhuewAQGlDl6iBeo,717
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v1/tp_model.py,sha256=MpZ2j3YyAQqQbyAHHNG8Mgn2sHKTRvVoBPQHMIiyG70,6310
-model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v1/tpc_keras.py,sha256=lEARkzeJxgUemSqssJ1KJeAuOmzJMfwrEJQqMZ9P_Ts,3845
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v1/tpc_keras.py,sha256=KHj1i_TwQkhDzUEbUm0HZxqQLCl57H6CsH8zWu02jJo,3830
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v1/tpc_pytorch.py,sha256=-_hZ1eHcDW2emKcCwkWkuYjqYaOZUmz5sSlIBOI83YA,3261
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v2/__init__.py,sha256=GyJf-kpSYJDM9lmnxf4JFrMvRj4K8K-4te5r9qHCus8,717
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v2/tp_model.py,sha256=E9h5L5wmHgzLPrFBCf8eXUtZAIQ_3zLMijQL6l7jKsE,6846
-model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v2/tpc_keras.py,sha256=FgXhVU9ZH00WLphJn5bneSjKVkauBdPmOI3DaO7s-d0,4321
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v2/tpc_keras.py,sha256=mq_KeCLBO0n0JfzA7LK8VPEngX5-FdK_NHmLZJIGD-w,4306
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v2/tpc_pytorch.py,sha256=ZV73uahc1wksuQWk5G5PmN3dgeELFXPNPDur8l14bJU,3859
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3/__init__.py,sha256=e9S6b6dxUdRX96B3ZLVkYEaxjFEWb-ml6v-14g_xbhg,717
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3/tp_model.py,sha256=hUtEUeq4hWMWGAA_I_oO1HhOs2CwWSLi3IqUGkutrR4,8148
-model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3/tpc_keras.py,sha256=XLr2wpUGH5B9CKfvhA-IaxEpB6PZzg5ZHMcXxr0668I,4618
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3/tpc_keras.py,sha256=iLSTsQs0v8KLW-AcSfwXVxKLY4vdrwawTtuBS17S-Wg,4603
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3/tpc_pytorch.py,sha256=R2GGyFFiR9qg9ycakJ3cZPpSDLcEtgpU91tARwWv7oc,4196
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3_lut/__init__.py,sha256=l8UGahYXmyI7UNH6VM0Q0Bzp0ncXIDRx-zNyyFEH1tc,721
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3_lut/tp_model.py,sha256=zkBRVigTdd25_0TpuvzQbZpKBH5PZa-h4MRUc19zE5A,8332
-model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3_lut/tpc_keras.py,sha256=HFJCal2BMsdVHUN_1686Ke8FQFSyuYodWriCXhogBIg,4631
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3_lut/tpc_keras.py,sha256=hAPEhWVk0oebcttbF_8-VyVens_8TjCtiazTguj4_sc,4616
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3_lut/tpc_pytorch.py,sha256=VtL9tQGsVy8oQwc3wM8a6r6lnvHVdk3E9gSmgmJobJU,4209
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4/__init__.py,sha256=5r1xy_YVlr9WJVola7SuvU2pt7B5KYi20Jz78M5uTPw,717
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4/tp_model.py,sha256=hUtEUeq4hWMWGAA_I_oO1HhOs2CwWSLi3IqUGkutrR4,8148
-model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4/tpc_keras.py,sha256=dZylTAVVQDf2DioiWIFiVmQwMfAQHmK7DXm_IM8Jjf4,5372
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4/tpc_keras.py,sha256=D-F6B50UDb5u3DxtcBSBTjvwlyzTwOYSBTzNwz11JkY,5367
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4/tpc_pytorch.py,sha256=mXu9X0Z5qotW12B4tTzFeuLHWUm8n9aCUEQA2CmdNy4,4714
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4_lut/__init__.py,sha256=G3K-LUGYaII8WGxZOXgyhT5ISpLMy-sV08Xn16PRcOw,721
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4_lut/tp_model.py,sha256=zkBRVigTdd25_0TpuvzQbZpKBH5PZa-h4MRUc19zE5A,8332
-model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4_lut/tpc_keras.py,sha256=OxJrh4qVLT-9ApSgQkC7WoP_0CXaJPYMTlkQ4e4R5eI,5541
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4_lut/tpc_keras.py,sha256=fLZC9GEyqMWQnZTLJzJe56Ujt3kr62L8zs7ygk50wPw,5536
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4_lut/tpc_pytorch.py,sha256=JifukO5o082ldp8tbOu3QJU4EnOyfW-LTjFuzY-VYJ0,4727
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v5/__init__.py,sha256=tafo2--3T8rFuwRr9nXlXGBfGTg7fCX7sGjh3hD5YXQ,717
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v5/tp_model.py,sha256=97CxWMRfb-Xutkl95V3Jy0DbEJomQ7vGfv8afunG26A,8145
-model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v5/tpc_keras.py,sha256=sLaLaGjdVT7HgU-5eQbs9ABA4i1vnR7yGAFStgo6r-E,5437
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v5/tpc_keras.py,sha256=cd__C4W4fKF7SJ9O43HFvQxbLG7z39hl9GNg3HjjY0w,5432
 model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v5/tpc_pytorch.py,sha256=676izRmvryL9ZbO60yWf0YmK58A988mr7YCIQWc6emQ,4714
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/__init__.py,sha256=lNJ29DYxaLUPDstRDA1PGI5r9Fulq_hvrZMlhst1Z5g,697
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/target_platform_capabilities.py,sha256=MvLXPoTaYacBFU0oNjdLq0bRT66rAKEN5MWrB6oDIyE,2295
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/latest/__init__.py,sha256=F5RG4MnuAwKcNXbfVbPFLQu30-lNax-7knqu20B6udQ,1522
 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=PhGlCGLcDN9m8yiPeNNxpUC9TiC95qWHx7WIlxcwXoA,8140
-model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1/tpc_keras.py,sha256=MqbFJxLYsiuDYyHAS-6H6s7L7_eN45mDyOlvCmhBpDU,5443
+model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1/tpc_keras.py,sha256=lDV2Wp_3w8mjHkEWRpA0l0zgr4I43PddLYbZ6NGvNoE,5439
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1/tpc_pytorch.py,sha256=4liNmjx5lKxlkekqRlkJ0CPUMlsFcuktLxkIBB8lrI0,4721
 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=81pYnxAU9di5AXzW1TTfP7rS0IJWd9hjrI2VRQzuEa0,8443
-model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1_lut/tpc_keras.py,sha256=fu2pnmuSD2U_GQwiETLQpMYvs332uks14gCAdJCS-eM,5451
+model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1_lut/tpc_keras.py,sha256=R9-utIQFQ5K89o7r7JSYlYvpibl9ZTN-uz6CWsfKvzE,5446
 model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1_lut/tpc_pytorch.py,sha256=CsRgw4_QapBQ7Y1_QQ-kI_RewooBwqyRsMrmlB_k-kc,4729
 model_compression_toolkit/target_platform_capabilities/tpc_models/qnnpack_tpc/__init__.py,sha256=cco4TmeIDIh32nj9ZZXVkws4dd9F2UDrmjKzTN8G0V0,697
 model_compression_toolkit/target_platform_capabilities/tpc_models/qnnpack_tpc/target_platform_capabilities.py,sha256=lnhJcwvTF0t7ybeiTleIS1p0aD8xzFZxVPx4ISk5uWQ,2090
 model_compression_toolkit/target_platform_capabilities/tpc_models/qnnpack_tpc/latest/__init__.py,sha256=UUvUCcTots_sehdRnDfgkaE8WPQ7dPbeuhDF4Qy2nzw,1510
 model_compression_toolkit/target_platform_capabilities/tpc_models/qnnpack_tpc/v1/__init__.py,sha256=t4JKsPcor-7KSCKzIwuaBv0NLNwfhuewAQGlDl6iBeo,717
 model_compression_toolkit/target_platform_capabilities/tpc_models/qnnpack_tpc/v1/tp_model.py,sha256=XOl5GO_BIDVhZFn4TX6v429Spq-JKUnEgCHzF8kY9TM,6204
-model_compression_toolkit/target_platform_capabilities/tpc_models/qnnpack_tpc/v1/tpc_keras.py,sha256=0Wsk6qyok8S0UBf7Io011D3pp3bRqJvjv74JdF2tKRw,3198
+model_compression_toolkit/target_platform_capabilities/tpc_models/qnnpack_tpc/v1/tpc_keras.py,sha256=rIhqfc_Ks1SKzk5ppfomP88fGhx3Oh2w498dSW3p-ZY,3183
 model_compression_toolkit/target_platform_capabilities/tpc_models/qnnpack_tpc/v1/tpc_pytorch.py,sha256=o5w_ylMr7KCJCbSP1jx8Qs8HFysEEg4e_aRM_LXlexM,2942
 model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/__init__.py,sha256=cco4TmeIDIh32nj9ZZXVkws4dd9F2UDrmjKzTN8G0V0,697
 model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/target_platform_capabilities.py,sha256=C42RJBQERjjkqFa1pRsHZXseofceOtrp_fJxs4wqWvo,2085
 model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/latest/__init__.py,sha256=sK9PnyB2R9g0rqHr_9vyUFX7wSyrZe7x9yqYUlbaiqo,1505
 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=u8KAytDvkXAbgqpVEKynSqag3Wrc7TAtCP9Ru9Y0hR8,8048
-model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/v1/tpc_keras.py,sha256=bce4YrSBmFk2NNVmL4_GysGn_Ejhgk4yHRXOAKZKv_I,5925
+model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/v1/tpc_keras.py,sha256=KeRMMdcqWvZ2MACbYe2Lg_M_RJ-VZkKWHni-GcDZBNs,5920
 model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/v1/tpc_pytorch.py,sha256=KH_kdxyX9pPeLqkIj9_E59ZhAIFZ8bll0sfwDuILVdU,5068
 model_compression_toolkit/trainable_infrastructure/__init__.py,sha256=O0J24JDzwdo-p9yf1oVtDnK1ZoZ1K8iPAp7WJ40bcTQ,1104
 model_compression_toolkit/trainable_infrastructure/common/__init__.py,sha256=huHoBUcKNB6BnY6YaUCcFvdyBtBI172ZoUD8ZYeNc6o,696
@@ -416,13 +416,13 @@ model_compression_toolkit/trainable_infrastructure/common/quant_utils.py,sha256=
 model_compression_toolkit/trainable_infrastructure/common/trainable_quantizer_config.py,sha256=My5Wz34jPOyh8z33OTpKnOobRB0cpO_Qgmtsd5lizHo,4791
 model_compression_toolkit/trainable_infrastructure/keras/__init__.py,sha256=huHoBUcKNB6BnY6YaUCcFvdyBtBI172ZoUD8ZYeNc6o,696
 model_compression_toolkit/trainable_infrastructure/keras/base_keras_quantizer.py,sha256=UJRsEBp32qVU34ICKPfn9Ow4OjLQn33GjKA_3C-9Y7M,4178
-model_compression_toolkit/trainable_infrastructure/keras/config_serialization.py,sha256=PpyM0u8fsvv5xw5Gmt-3Z1UFATX5KLJzLbsXaquuyuk,3991
+model_compression_toolkit/trainable_infrastructure/keras/config_serialization.py,sha256=txdWXdZoHazg-3MDPb9P-oXRM92LRn2G_8woEplwKaI,4360
 model_compression_toolkit/trainable_infrastructure/keras/load_model.py,sha256=DwpVuJsTf-Sp3B-7vYuhQdFXPtVopNGRcanL-WXzEzw,3446
 model_compression_toolkit/trainable_infrastructure/keras/quantizer_utils.py,sha256=MVwXNymmFRB2NXIBx4e2mdJ1RfoHxRPYRgjb1MQP5kY,1797
 model_compression_toolkit/trainable_infrastructure/pytorch/__init__.py,sha256=huHoBUcKNB6BnY6YaUCcFvdyBtBI172ZoUD8ZYeNc6o,696
 model_compression_toolkit/trainable_infrastructure/pytorch/base_pytorch_quantizer.py,sha256=SbvRlIdE32PEBsINt1bhSqvrKL_zbM9V-aeSkOn-sw4,3083
-mct_nightly-1.9.0.20230814.post352.dist-info/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
-mct_nightly-1.9.0.20230814.post352.dist-info/METADATA,sha256=NqhVoJcRgeb3DDNnTzIr4lTEk7fqgP4uMShzIt9b2dg,10750
-mct_nightly-1.9.0.20230814.post352.dist-info/WHEEL,sha256=5sUXSg9e4bi7lTLOHcm6QEYwO5TIF1TNbTSVFVjcJcc,92
-mct_nightly-1.9.0.20230814.post352.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
-mct_nightly-1.9.0.20230814.post352.dist-info/RECORD,,
+mct_nightly-1.9.0.20230816.post359.dist-info/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
+mct_nightly-1.9.0.20230816.post359.dist-info/METADATA,sha256=_Cg3_JlV6RcNQaK_2acApauZ64GM0-E-2KbsPBhZjTg,10750
+mct_nightly-1.9.0.20230816.post359.dist-info/WHEEL,sha256=5sUXSg9e4bi7lTLOHcm6QEYwO5TIF1TNbTSVFVjcJcc,92
+mct_nightly-1.9.0.20230816.post359.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
+mct_nightly-1.9.0.20230816.post359.dist-info/RECORD,,

model_compression_toolkit/core/keras/back2framework/keras_model_builder.py

@@ -20,12 +20,10 @@ from packaging import version
 from model_compression_toolkit.core.common.back2framework.base_model_builder import BaseModelBuilder
 from model_compression_toolkit.core.common.user_info import UserInformation
 
-# As from Tensorflow 2.6, keras is a separate package and some classes should be imported differently.
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.keras.layers import Input
-    from tensorflow.python.keras.layers.core import TFOpLambda
-    from tensorflow.python.keras.engine.base_layer import TensorFlowOpLayer
-    from tensorflow.python.keras.layers import Layer
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras import Input
+    from keras.src.layers.core import TFOpLambda
+    from keras.src.engine.base_layer import TensorFlowOpLayer, Layer
 else:
     from keras import Input
     from keras.layers.core import TFOpLambda

model_compression_toolkit/core/keras/back2framework/mixed_precision_model_builder.py

@@ -14,7 +14,13 @@
 # ==============================================================================
 from typing import Tuple, Any, Dict, Union, List
 
-from keras.engine.base_layer import Layer
+from packaging import version
+import tensorflow as tf
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.engine.base_layer import Layer
+else:
+    from keras.engine.base_layer import Layer
+
 from keras.models import Model
 from mct_quantizers import KerasQuantizationWrapper, KerasActivationQuantizationHolder, QuantizationTarget
 from mct_quantizers.common.get_quantizers import get_inferable_quantizer_class

model_compression_toolkit/core/keras/back2framework/model_gradients.py

@@ -19,10 +19,10 @@ from packaging import version
 # As from Tensorflow 2.6, keras is a separate package and some classes should be imported differently.
 from tqdm import tqdm
 
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.python.keras.layers import Layer  # pragma: no cover
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.engine.base_layer import Layer
 else:
-    from keras.engine.base_layer import Layer
+    from tensorflow.python.keras.engine.base_layer import Layer
 
 from typing import Any, Dict, List, Tuple
 from tensorflow.python.util.object_identity import Reference as TFReference

model_compression_toolkit/core/keras/default_framework_info.py

@@ -19,8 +19,8 @@ import tensorflow as tf
 from model_compression_toolkit.core.keras.quantizer.lut_fake_quant import activation_lut_kmean_quantizer
 from packaging import version
 
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.keras.layers import Conv2D, DepthwiseConv2D, Dense, Conv2DTranspose, Softmax, ELU
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.layers import Conv2D, DepthwiseConv2D, Dense, Conv2DTranspose, Softmax, ELU
 else:
     from keras.layers import Conv2D, DepthwiseConv2D, Dense, Conv2DTranspose, Softmax, ELU
 

model_compression_toolkit/core/keras/graph_substitutions/substitutions/multi_head_attention_decomposition.py

@@ -16,9 +16,9 @@ import numpy as np
 import tensorflow as tf
 from packaging import version
 
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.python.keras.layers.core import TFOpLambda
-    from tensorflow.keras.layers import MultiHeadAttention, Conv2D, Softmax, Concatenate, Reshape, Permute
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.layers.core import TFOpLambda
+    from keras.src.layers import MultiHeadAttention, Conv2D, Softmax, Concatenate, Reshape, Permute
 else:
     from keras.layers.core import TFOpLambda
     from keras.layers import MultiHeadAttention, Conv2D, Softmax, Concatenate, Reshape, Permute

model_compression_toolkit/core/keras/graph_substitutions/substitutions/shift_negative_activation.py

@@ -16,11 +16,16 @@
 from typing import Tuple, Any
 
 import numpy as np
-import tensorflow as tf
 
+from packaging import version
+import tensorflow as tf
 from tensorflow.python.keras.layers.core import TFOpLambda
-from tensorflow.keras.layers import Activation, Conv2D, Dense, DepthwiseConv2D, ZeroPadding2D, Reshape, \
-    GlobalAveragePooling2D, Dropout, ReLU, PReLU, ELU
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.layers import Activation, Conv2D, Dense, DepthwiseConv2D, ZeroPadding2D, Reshape, \
+        GlobalAveragePooling2D, Dropout, ReLU, PReLU, ELU
+else:
+    from tensorflow.keras.layers import Activation, Conv2D, Dense, DepthwiseConv2D, ZeroPadding2D, Reshape, \
+        GlobalAveragePooling2D, Dropout, ReLU, PReLU, ELU
 
 from model_compression_toolkit.core import CoreConfig, FrameworkInfo
 from model_compression_toolkit.core.common import BaseNode, Graph

model_compression_toolkit/core/keras/keras_implementation.py

@@ -41,9 +41,10 @@ from model_compression_toolkit.core.keras.statistics_correction.apply_second_mom
     keras_apply_second_moment_correction
 from packaging import version
 
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.keras.layers import Dense, Activation, Conv2D, DepthwiseConv2D, Conv2DTranspose, Concatenate, Add
-    from tensorflow.python.keras.layers.core import TFOpLambda
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.layers import Dense, Activation, Conv2D, DepthwiseConv2D, Conv2DTranspose, \
+        Concatenate, Add
+    from keras.src.layers.core import TFOpLambda
 else:
     from keras.layers import Dense, Activation, Conv2D, DepthwiseConv2D, Conv2DTranspose, \
         Concatenate, Add

model_compression_toolkit/core/keras/keras_node_prior_info.py

@@ -3,8 +3,8 @@ import numpy as np
 import tensorflow as tf
 from packaging import version
 
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.keras.layers import Activation, ReLU, BatchNormalization
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.layers import Activation, ReLU, BatchNormalization
 else:
     from keras.layers import Activation, ReLU, BatchNormalization
 

model_compression_toolkit/core/keras/reader/common.py

@@ -17,12 +17,12 @@
 import tensorflow as tf
 from packaging import version
 
-# As from Tensorflow 2.6, keras is a separate package and some classes should be imported differently.
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.python.keras.engine.node import Node as KerasNode
-    from tensorflow.keras.layers import InputLayer
-    from tensorflow.python.keras.engine.functional import Functional
-    from tensorflow.python.keras.engine.sequential import Sequential
+
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.engine.input_layer import InputLayer
+    from keras.src.engine.node import Node as KerasNode
+    from keras.src.engine.functional import Functional
+    from keras.src.engine.sequential import Sequential
 else:
     from keras.engine.input_layer import InputLayer
     from keras.engine.node import Node as KerasNode

model_compression_toolkit/core/keras/reader/connectivity_handler.py

@@ -17,9 +17,8 @@
 import tensorflow as tf
 from packaging import version
 
-# As from Tensorflow 2.6, keras is a separate package and some classes should be imported differently.
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.python.keras.engine.node import Node as KerasNode
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.engine.node import Node as KerasNode
 else:
     from keras.engine.node import Node as KerasNode
 

model_compression_toolkit/core/keras/reader/node_builder.py

@@ -20,10 +20,10 @@ from packaging import version
 from model_compression_toolkit.core.keras.custom_layer_validation import is_keras_custom_layer
 from model_compression_toolkit.logger import Logger
 
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.python.keras.layers.core import TFOpLambda, SlicingOpLambda
-    from tensorflow.python.keras.engine.keras_tensor import KerasTensor
-    from tensorflow.python.keras.engine.node import Node as KerasNode
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.layers.core import TFOpLambda, SlicingOpLambda
+    from keras.src.engine.keras_tensor import KerasTensor
+    from keras.src.engine.node import Node as KerasNode
 else:
     from keras.layers.core import TFOpLambda, SlicingOpLambda
     from keras.engine.keras_tensor import KerasTensor

model_compression_toolkit/exporter/model_exporter/keras/fakely_quant_keras_exporter.py

@@ -18,7 +18,11 @@ from typing import Dict, Callable
 
 import keras.models
 import tensorflow as tf
-from keras.engine.base_layer import Layer
+from packaging import version
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.engine.base_layer import Layer
+else:
+    from keras.engine.base_layer import Layer
 
 from mct_quantizers import KerasQuantizationWrapper
 from mct_quantizers.logger import Logger
@@ -27,8 +31,8 @@ layers = keras.layers
 
 from model_compression_toolkit.exporter.model_exporter.keras.base_keras_exporter import BaseKerasExporter
 
-from keras.engine import base_layer_utils
-from keras import backend
+from tensorflow.python.keras.engine import base_layer_utils
+from tensorflow.python.keras import backend
 
 
 class FakelyQuantKerasExporter(BaseKerasExporter):

model_compression_toolkit/exporter/model_wrapper/keras/validate_layer.py

@@ -19,8 +19,15 @@ from model_compression_toolkit.constants import FOUND_TF
 from model_compression_toolkit.logger import Logger
 
 if FOUND_TF:
-    from keras.engine.base_layer import Layer
-    from keras.engine.input_layer import InputLayer
+    from packaging import version
+    import tensorflow as tf
+    if version.parse(tf.__version__) >= version.parse("2.13"):
+        from keras.src.engine.base_layer import Layer
+        from keras.src.engine.input_layer import InputLayer
+    else:
+        from keras.engine.base_layer import Layer
+        from keras.engine.input_layer import InputLayer
+
     from mct_quantizers import KerasQuantizationWrapper
 
     def is_keras_layer_exportable(layer: Any) -> bool:

model_compression_toolkit/exporter/model_wrapper/pytorch/builder/node_to_quantizer.py

@@ -51,15 +51,15 @@ def get_weights_inferable_quantizer_kwargs(node_qc: NodeWeightsQuantizationConfi
     if quantization_method in [QuantizationMethod.POWER_OF_TWO,
                                QuantizationMethod.SYMMETRIC]:
         return {qi_inferable_quantizers_constants.NUM_BITS: node_qc.weights_n_bits,
-                qi_inferable_quantizers_constants.THRESHOLD: node_qc.weights_quantization_params[THRESHOLD].flatten(),
+                qi_inferable_quantizers_constants.THRESHOLD: node_qc.weights_quantization_params[THRESHOLD].flatten().tolist(),
                 qi_inferable_quantizers_constants.PER_CHANNEL: node_qc.weights_per_channel_threshold,
                 qi_inferable_quantizers_constants.CHANNEL_AXIS: node_qc.weights_channels_axis}
 
     elif quantization_method in [QuantizationMethod.UNIFORM]:
         return {qi_inferable_quantizers_constants.NUM_BITS: node_qc.weights_n_bits,
                 qi_inferable_quantizers_constants.PER_CHANNEL: node_qc.weights_per_channel_threshold,
-                qi_inferable_quantizers_constants.MIN_RANGE: node_qc.weights_quantization_params[RANGE_MIN].flatten(),
-                qi_inferable_quantizers_constants.MAX_RANGE: node_qc.weights_quantization_params[RANGE_MAX].flatten(),
+                qi_inferable_quantizers_constants.MIN_RANGE: node_qc.weights_quantization_params[RANGE_MIN].flatten().tolist(),
+                qi_inferable_quantizers_constants.MAX_RANGE: node_qc.weights_quantization_params[RANGE_MAX].flatten().tolist(),
                 qi_inferable_quantizers_constants.CHANNEL_AXIS: node_qc.weights_channels_axis}
 
     elif quantization_method in [QuantizationMethod.LUT_POT_QUANTIZER, QuantizationMethod.LUT_SYM_QUANTIZER]:
@@ -96,13 +96,13 @@ def get_activation_inferable_quantizer_kwargs(node_qc: NodeActivationQuantizatio
     if quantization_method in [QuantizationMethod.POWER_OF_TWO,
                                QuantizationMethod.SYMMETRIC]:
         return {qi_inferable_quantizers_constants.NUM_BITS: node_qc.activation_n_bits,
-                qi_inferable_quantizers_constants.THRESHOLD: np.asarray([node_qc.activation_quantization_params[THRESHOLD]]),
+                qi_inferable_quantizers_constants.THRESHOLD: [node_qc.activation_quantization_params[THRESHOLD]],
                 qi_inferable_quantizers_constants.SIGNED: node_qc.activation_quantization_params.get(SIGNED)}
 
     elif quantization_method in [QuantizationMethod.UNIFORM]:
         return {qi_inferable_quantizers_constants.NUM_BITS: node_qc.activation_n_bits,
-                qi_inferable_quantizers_constants.MIN_RANGE: np.asarray([node_qc.activation_quantization_params[RANGE_MIN]]),
-                qi_inferable_quantizers_constants.MAX_RANGE: np.asarray([node_qc.activation_quantization_params[RANGE_MAX]])}
+                qi_inferable_quantizers_constants.MIN_RANGE: [node_qc.activation_quantization_params[RANGE_MIN]],
+                qi_inferable_quantizers_constants.MAX_RANGE: [node_qc.activation_quantization_params[RANGE_MAX]]}
 
     elif quantization_method in [QuantizationMethod.LUT_POT_QUANTIZER]:
         return {qi_inferable_quantizers_constants.NUM_BITS: node_qc.activation_n_bits,

model_compression_toolkit/gptq/keras/gptq_training.py

@@ -28,8 +28,8 @@ from model_compression_toolkit.gptq.keras.quantizer.quantization_builder import
 from model_compression_toolkit.logger import Logger
 from mct_quantizers import KerasQuantizationWrapper, KerasActivationQuantizationHolder
 
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.python.keras.engine.base_layer import TensorFlowOpLayer
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.engine.base_layer import TensorFlowOpLayer
 else:
     from keras.engine.base_layer import TensorFlowOpLayer
 

model_compression_toolkit/qat/pytorch/quantizer/ste_rounding/symmetric_ste.py

@@ -119,12 +119,12 @@ class STEWeightQATQuantizer(BasePytorchQATTrainableQuantizer):
         if self.power_of_two:
             pot_threshold = 2 ** np.ceil(np.log2(np_threshold))
             return WeightsPOTInferableQuantizer(num_bits=self.num_bits,
-                                                threshold=pot_threshold,
+                                                threshold=pot_threshold.tolist(),
                                                 per_channel=self.quantization_config.weights_per_channel_threshold,
                                                 channel_axis=self.quantization_config.weights_channels_axis)
         else:
             return WeightsSymmetricInferableQuantizer(num_bits=self.num_bits,
-                                                      threshold=np_threshold,
+                                                      threshold=np_threshold.tolist(),
                                                       per_channel=self.quantization_config.weights_per_channel_threshold,
                                                       channel_axis=self.quantization_config.weights_channels_axis)
 
@@ -198,9 +198,9 @@ class STEActivationQATQuantizer(BasePytorchQATTrainableQuantizer):
         if self.power_of_two:
             pot_threshold = np.power(2.0, np.ceil(np.log2(np_threshold)))
             return ActivationPOTInferableQuantizer(num_bits=self.num_bits,
-                                                   threshold=pot_threshold,
+                                                   threshold=pot_threshold.tolist(),
                                                    signed=self.sign)
         else:
             return ActivationSymmetricInferableQuantizer(num_bits=self.num_bits,
-                                                         threshold=np_threshold,
+                                                         threshold=np_threshold.tolist(),
                                                          signed=self.sign)

model_compression_toolkit/qat/pytorch/quantizer/ste_rounding/uniform_ste.py

@@ -190,4 +190,5 @@ class STEUniformActivationQATQuantizer(BasePytorchQATTrainableQuantizer):
         _max = self.get_quantizer_variable(FQ_MAX).cpu().detach().numpy()
 
         return ActivationUniformInferableQuantizer(num_bits=self.num_bits,
-                                                   min_range=_min, max_range=_max)
+                                                   min_range=_min.tolist(),
+                                                   max_range=_max.tolist())

model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v1/tpc_keras.py

@@ -15,9 +15,8 @@
 import tensorflow as tf
 from packaging import version
 
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.keras.layers import Conv2D, DepthwiseConv2D, Reshape, ZeroPadding2D, \
-        Dropout, \
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.layers import Conv2D, DepthwiseConv2D, Reshape, ZeroPadding2D, Dropout, \
         MaxPooling2D, Activation, ReLU, Flatten, Cropping2D
 else:
     from keras.layers import Conv2D, DepthwiseConv2D, Reshape, ZeroPadding2D, \

model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v2/tpc_keras.py

@@ -16,10 +16,9 @@ import tensorflow as tf
 
 from packaging import version
 
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.keras.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, \
-        Dropout, \
-        MaxPooling2D, Activation, ReLU, PReLU, Flatten, Cropping2D
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, \
+        Dropout, MaxPooling2D, Activation, ReLU, PReLU, Flatten, Cropping2D
 else:
     from keras.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, \
         Dropout, MaxPooling2D, Activation, ReLU, PReLU, Flatten, Cropping2D

model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3/tpc_keras.py

@@ -15,10 +15,9 @@
 import tensorflow as tf
 from packaging import version
 
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.keras.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, \
-        Dropout, \
-        MaxPooling2D, Activation, ReLU, Add, Subtract, Multiply, PReLU, Flatten, Cropping2D
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, \
+        Dropout, MaxPooling2D, Activation, ReLU, Add, Subtract, Multiply, PReLU, Flatten, Cropping2D
 else:
     from keras.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, \
         Dropout, MaxPooling2D, Activation, ReLU, Add, Subtract, Multiply, PReLU, Flatten, Cropping2D

model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3_lut/tpc_keras.py

@@ -16,10 +16,9 @@ import tensorflow as tf
 from packaging import version
 from model_compression_toolkit.target_platform_capabilities.tpc_models.default_tpc.v3_lut import __version__ as TPC_VERSION
 
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.keras.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, \
-        Dropout, \
-        MaxPooling2D, Activation, ReLU, Add, Subtract, Multiply, PReLU, Flatten, Cropping2D
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, \
+        Dropout, MaxPooling2D, Activation, ReLU, Add, Subtract, Multiply, PReLU, Flatten, Cropping2D
 else:
     from keras.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, \
         Dropout, MaxPooling2D, Activation, ReLU, Add, Subtract, Multiply, PReLU, Flatten, Cropping2D

model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4/tpc_keras.py

@@ -15,8 +15,8 @@
 import tensorflow as tf
 from packaging import version
 
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.keras.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, Dropout, \
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, Dropout, \
         MaxPooling2D, Activation, ReLU, Add, Subtract, Multiply, PReLU, Flatten, Cropping2D, LeakyReLU, Permute, \
         Conv2DTranspose
 else:

model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4_lut/tpc_keras.py

@@ -16,8 +16,8 @@ import tensorflow as tf
 from packaging import version
 from model_compression_toolkit.target_platform_capabilities.tpc_models.default_tpc.v4_lut import __version__ as TPC_VERSION
 
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.keras.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, Dropout, \
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, Dropout, \
         MaxPooling2D, Activation, ReLU, Add, Subtract, Multiply, PReLU, Flatten, Cropping2D, LeakyReLU, Permute, \
         Conv2DTranspose
 else:

model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v5/tpc_keras.py

@@ -15,8 +15,8 @@
 import tensorflow as tf
 from packaging import version
 
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.keras.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, Dropout, \
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, Dropout, \
         MaxPooling2D, Activation, ReLU, Add, Subtract, Multiply, PReLU, Flatten, Cropping2D, LeakyReLU, Permute, \
         Conv2DTranspose
 else:

model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1/tpc_keras.py

@@ -15,8 +15,9 @@
 import tensorflow as tf
 from packaging import version
 
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.keras.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, Dropout, \
+
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, Dropout, \
         MaxPooling2D, Activation, ReLU, Add, Subtract, Multiply, PReLU, Flatten, Cropping2D, LeakyReLU, Permute, \
         Conv2DTranspose
 else:

model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1_lut/tpc_keras.py

@@ -15,8 +15,8 @@
 import tensorflow as tf
 from packaging import version
 
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.keras.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, Dropout, \
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.layers import Conv2D, DepthwiseConv2D, Dense, Reshape, ZeroPadding2D, Dropout, \
         MaxPooling2D, Activation, ReLU, Add, Subtract, Multiply, PReLU, Flatten, Cropping2D, LeakyReLU, Permute, \
         Conv2DTranspose
 else:

model_compression_toolkit/target_platform_capabilities/tpc_models/qnnpack_tpc/v1/tpc_keras.py

@@ -17,9 +17,8 @@ import tensorflow as tf
 from packaging import version
 from model_compression_toolkit.target_platform_capabilities.tpc_models.qnnpack_tpc.v1 import __version__ as TPC_VERSION
 
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.keras.layers import Conv2D, DepthwiseConv2D, Conv2DTranspose, Dense, BatchNormalization, ReLU, \
-        Activation
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.layers import Conv2D, DepthwiseConv2D, Conv2DTranspose, Dense, BatchNormalization, ReLU, Activation
 else:
     from keras.layers import Conv2D, DepthwiseConv2D, Conv2DTranspose, Dense, BatchNormalization, ReLU, Activation
 

model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/v1/tpc_keras.py

@@ -15,9 +15,9 @@
 import tensorflow as tf
 from packaging import version
 
-if version.parse(tf.__version__) < version.parse("2.6"):
-    from tensorflow.keras.layers import Conv2D, Dense, Reshape, ZeroPadding2D, AveragePooling2D, Activation, \
-        DepthwiseConv2D, MaxPooling2D, ReLU, Add, Softmax, Concatenate, Multiply, Maximum, Minimum, BatchNormalization
+if version.parse(tf.__version__) >= version.parse("2.13"):
+    from keras.src.layers import Conv2D, Dense, Reshape, ZeroPadding2D, AveragePooling2D, Activation, DepthwiseConv2D, \
+        MaxPooling2D, ReLU, Add, Softmax, Concatenate, Multiply, Maximum, Minimum, BatchNormalization
 else:
     from keras.layers import Conv2D, Dense, Reshape, ZeroPadding2D, AveragePooling2D, Activation, DepthwiseConv2D, \
         MaxPooling2D, ReLU, Add, Softmax, Concatenate, Multiply, Maximum, Minimum, BatchNormalization

model_compression_toolkit/trainable_infrastructure/keras/config_serialization.py

@@ -17,12 +17,17 @@ import copy
 from typing import Any, Union
 from enum import Enum
 
+import numpy as np
+
 from model_compression_toolkit.target_platform_capabilities.target_platform import QuantizationMethod
 from model_compression_toolkit.trainable_infrastructure.common.trainable_quantizer_config import \
     TrainableQuantizerActivationConfig, TrainableQuantizerWeightsConfig
 from mct_quantizers.common import constants as C
 
 
+CONFIG = "config"
+VALUE = "value"
+
 def transform_enum(v: Any):
     """
     If an enum is received it value is return otherwise the input is returned.
@@ -63,9 +68,13 @@ def config_deserialization(in_config: dict) -> Union[TrainableQuantizerWeightsCo
     """
     in_config = copy.deepcopy(in_config)
     if in_config[C.IS_WEIGHTS]:
+        weights_quantization_params = {}
+        for key, value in in_config[C.WEIGHTS_QUANTIZATION_PARAMS].items():
+            # In TF2.13.0, serialization of numpy array is dictionary with parameters
+            weights_quantization_params.update({key: np.array(value[CONFIG][VALUE] if isinstance(value, dict) else value)})
         return TrainableQuantizerWeightsConfig(weights_quantization_method=QuantizationMethod(in_config[C.WEIGHTS_QUANTIZATION_METHOD]),
                                                weights_n_bits=in_config[C.WEIGHTS_N_BITS],
-                                               weights_quantization_params=in_config[C.WEIGHTS_QUANTIZATION_PARAMS],
+                                               weights_quantization_params=weights_quantization_params,
                                                enable_weights_quantization=in_config[C.ENABLE_WEIGHTS_QUANTIZATION],
                                                weights_channels_axis=in_config[C.WEIGHTS_CHANNELS_AXIS],
                                                weights_per_channel_threshold=in_config[C.WEIGHTS_PER_CHANNEL_THRESHOLD],