mct-nightly 1.8.0.8032023.post421__tar.gz → 1.8.0.8052023.post414__tar.gz

{mct-nightly-1.8.0.8032023.post421 → mct-nightly-1.8.0.8052023.post414}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mct-nightly
-Version: 1.8.0.8032023.post421
+Version: 1.8.0.8052023.post414
 Summary: A Model Compression Toolkit for neural networks
 Home-page: UNKNOWN
 License: UNKNOWN
@@ -29,8 +29,8 @@ Description: # Model Compression Toolkit (MCT)
         ## Supported Features
         
         MCT supports different quantization methods:
-        * Post training quantization (PTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_post_training_quantization_experimental.html#ug-keras-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_post_training_quantization_experimental.html#ug-pytorch-post-training-quantization-experimental)
-        * Gradient-based post training quantization (GPTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_gradient_post_training_quantization_experimental.html#ug-keras-gradient-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_gradient_post_training_quantization_experimental.html#ug-pytorch-gradient-post-training-quantization-experimental)
+        * Post-training quantization (PTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_post_training_quantization_experimental.html#ug-keras-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_post_training_quantization_experimental.html#ug-pytorch-post-training-quantization-experimental)
+        * Gradient-based post-training quantization (GPTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_gradient_post_training_quantization_experimental.html#ug-keras-gradient-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_gradient_post_training_quantization_experimental.html#ug-pytorch-gradient-post-training-quantization-experimental)
         * Quantization aware training (QAT)[*](#experimental-features)
         
         
@@ -46,15 +46,16 @@ Description: # Model Compression Toolkit (MCT)
         * Symmetric
         * Uniform
         
-        Core features:
+        Main features:
         * <ins>Graph optimizations:</ins> Transforming the model to an equivalent (yet, more efficient) model (for example, batch-normalization layer folding to its preceding linear layer).
-        * <ins>Quantization parameter search:</ins> Different methods can be used to minimize the expected added quantization-noise during thresholds search (by default, we use Mean-Square-Errorm but other metrics can be used such as No-Clipping, Mean-Average-Error, and more).
+        * <ins>Quantization parameter search:</ins> Different methods can be used to minimize the expected added quantization-noise during thresholds search (by default, we use Mean-Square-Error, but other metrics can be used such as No-Clipping, Mean-Average-Error, and more).
         * <ins>Advanced quantization algorithms:</ins> To prevent a performance degradation some algorithms are applied such as: 
           * <ins>Shift negative correction:</ins> Symmetric activation quantization can hurt the model's performance when some layers output both negative and positive activations, but their range is asymmetric. For more details please visit [1].
           * <ins>Outliers filtering:</ins> Computing z-score for activation statistics to detect and remove outliers.
         * <ins>Clustering:</ins> Using non-uniform quantization grid to quantize the weights and activations to match their distributions.[*](#experimental-features)
         * <ins>Mixed-precision search:</ins> Assigning quantization bit-width per layer (for weights/activations), based on the layer's sensitivity to different bit-widths.
         * <ins>Visualization:</ins> You can use TensorBoard to observe useful information for troubleshooting the quantized model's performance (for example, the model in different phases of the quantization, collected statistics, similarity between layers of the float and quantized model and bit-width configuration for mixed-precision quantization). For more details, please read the [visualization documentation](https://sony.github.io/model_optimization/docs/guidelines/visualization.html).   
+        * <ins>Target Platform Capabilities:</ins> The Target Platform Capabilities (TPC) describes the target platform (an edge device with dedicated hardware). For more details, please read the [TPC README](model_compression_toolkit/target_platform_capabilities/README.md).   
         
         
         #### Experimental features 
@@ -87,15 +88,15 @@ Description: # Model Compression Toolkit (MCT)
         pip install mct-nightly
         ```
         
-        ### Requierments
+        ### Requirements
         
-        To run MCT, one of the supported frameworks, Tenosflow/Pytorch, needs to be installed.
+        To run MCT, one of the supported frameworks, Tensorflow/Pytorch, needs to be installed.
         
-        For using with Tensorflow please install the packages: 
+        For use with Tensorflow please install the packages: 
         [tensorflow](https://www.tensorflow.org/install), 
         [tensorflow-model-optimization](https://www.tensorflow.org/model_optimization/guide/install)
         
-        For using with PyTorch please install the packages: 
+        For use with PyTorch please install the packages: 
         [torch](https://pytorch.org/)
         
         Also, a [requirements](requirements.txt) file can be used to set up your environment.

{mct-nightly-1.8.0.8032023.post421 → mct-nightly-1.8.0.8052023.post414}/README.md

@@ -23,8 +23,8 @@ MCT is developed by researchers and engineers working at Sony Semiconductor Isra
 ## Supported Features
 
 MCT supports different quantization methods:
-* Post training quantization (PTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_post_training_quantization_experimental.html#ug-keras-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_post_training_quantization_experimental.html#ug-pytorch-post-training-quantization-experimental)
-* Gradient-based post training quantization (GPTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_gradient_post_training_quantization_experimental.html#ug-keras-gradient-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_gradient_post_training_quantization_experimental.html#ug-pytorch-gradient-post-training-quantization-experimental)
+* Post-training quantization (PTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_post_training_quantization_experimental.html#ug-keras-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_post_training_quantization_experimental.html#ug-pytorch-post-training-quantization-experimental)
+* Gradient-based post-training quantization (GPTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_gradient_post_training_quantization_experimental.html#ug-keras-gradient-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_gradient_post_training_quantization_experimental.html#ug-pytorch-gradient-post-training-quantization-experimental)
 * Quantization aware training (QAT)[*](#experimental-features)
 
 
@@ -40,15 +40,16 @@ In addition, MCT supports different quantization schemes for quantizing weights
 * Symmetric
 * Uniform
 
-Core features:
+Main features:
 * <ins>Graph optimizations:</ins> Transforming the model to an equivalent (yet, more efficient) model (for example, batch-normalization layer folding to its preceding linear layer).
-* <ins>Quantization parameter search:</ins> Different methods can be used to minimize the expected added quantization-noise during thresholds search (by default, we use Mean-Square-Errorm but other metrics can be used such as No-Clipping, Mean-Average-Error, and more).
+* <ins>Quantization parameter search:</ins> Different methods can be used to minimize the expected added quantization-noise during thresholds search (by default, we use Mean-Square-Error, but other metrics can be used such as No-Clipping, Mean-Average-Error, and more).
 * <ins>Advanced quantization algorithms:</ins> To prevent a performance degradation some algorithms are applied such as: 
   * <ins>Shift negative correction:</ins> Symmetric activation quantization can hurt the model's performance when some layers output both negative and positive activations, but their range is asymmetric. For more details please visit [1].
   * <ins>Outliers filtering:</ins> Computing z-score for activation statistics to detect and remove outliers.
 * <ins>Clustering:</ins> Using non-uniform quantization grid to quantize the weights and activations to match their distributions.[*](#experimental-features)
 * <ins>Mixed-precision search:</ins> Assigning quantization bit-width per layer (for weights/activations), based on the layer's sensitivity to different bit-widths.
 * <ins>Visualization:</ins> You can use TensorBoard to observe useful information for troubleshooting the quantized model's performance (for example, the model in different phases of the quantization, collected statistics, similarity between layers of the float and quantized model and bit-width configuration for mixed-precision quantization). For more details, please read the [visualization documentation](https://sony.github.io/model_optimization/docs/guidelines/visualization.html).   
+* <ins>Target Platform Capabilities:</ins> The Target Platform Capabilities (TPC) describes the target platform (an edge device with dedicated hardware). For more details, please read the [TPC README](model_compression_toolkit/target_platform_capabilities/README.md).   
 
 
 #### Experimental features 
@@ -81,15 +82,15 @@ A nightly package is also available (unstable):
 pip install mct-nightly
 ```
 
-### Requierments
+### Requirements
 
-To run MCT, one of the supported frameworks, Tenosflow/Pytorch, needs to be installed.
+To run MCT, one of the supported frameworks, Tensorflow/Pytorch, needs to be installed.
 
-For using with Tensorflow please install the packages: 
+For use with Tensorflow please install the packages: 
 [tensorflow](https://www.tensorflow.org/install), 
 [tensorflow-model-optimization](https://www.tensorflow.org/model_optimization/guide/install)
 
-For using with PyTorch please install the packages: 
+For use with PyTorch please install the packages: 
 [torch](https://pytorch.org/)
 
 Also, a [requirements](requirements.txt) file can be used to set up your environment.

{mct-nightly-1.8.0.8032023.post421 → mct-nightly-1.8.0.8052023.post414}/mct_nightly.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mct-nightly
-Version: 1.8.0.8032023.post421
+Version: 1.8.0.8052023.post414
 Summary: A Model Compression Toolkit for neural networks
 Home-page: UNKNOWN
 License: UNKNOWN
@@ -29,8 +29,8 @@ Description: # Model Compression Toolkit (MCT)
         ## Supported Features
         
         MCT supports different quantization methods:
-        * Post training quantization (PTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_post_training_quantization_experimental.html#ug-keras-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_post_training_quantization_experimental.html#ug-pytorch-post-training-quantization-experimental)
-        * Gradient-based post training quantization (GPTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_gradient_post_training_quantization_experimental.html#ug-keras-gradient-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_gradient_post_training_quantization_experimental.html#ug-pytorch-gradient-post-training-quantization-experimental)
+        * Post-training quantization (PTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_post_training_quantization_experimental.html#ug-keras-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_post_training_quantization_experimental.html#ug-pytorch-post-training-quantization-experimental)
+        * Gradient-based post-training quantization (GPTQ): [Keras API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/keras_gradient_post_training_quantization_experimental.html#ug-keras-gradient-post-training-quantization-experimental), [PyTorch API](https://sony.github.io/model_optimization/docs/api/experimental_api_docs/methods/pytorch_gradient_post_training_quantization_experimental.html#ug-pytorch-gradient-post-training-quantization-experimental)
         * Quantization aware training (QAT)[*](#experimental-features)
         
         
@@ -46,15 +46,16 @@ Description: # Model Compression Toolkit (MCT)
         * Symmetric
         * Uniform
         
-        Core features:
+        Main features:
         * <ins>Graph optimizations:</ins> Transforming the model to an equivalent (yet, more efficient) model (for example, batch-normalization layer folding to its preceding linear layer).
-        * <ins>Quantization parameter search:</ins> Different methods can be used to minimize the expected added quantization-noise during thresholds search (by default, we use Mean-Square-Errorm but other metrics can be used such as No-Clipping, Mean-Average-Error, and more).
+        * <ins>Quantization parameter search:</ins> Different methods can be used to minimize the expected added quantization-noise during thresholds search (by default, we use Mean-Square-Error, but other metrics can be used such as No-Clipping, Mean-Average-Error, and more).
         * <ins>Advanced quantization algorithms:</ins> To prevent a performance degradation some algorithms are applied such as: 
           * <ins>Shift negative correction:</ins> Symmetric activation quantization can hurt the model's performance when some layers output both negative and positive activations, but their range is asymmetric. For more details please visit [1].
           * <ins>Outliers filtering:</ins> Computing z-score for activation statistics to detect and remove outliers.
         * <ins>Clustering:</ins> Using non-uniform quantization grid to quantize the weights and activations to match their distributions.[*](#experimental-features)
         * <ins>Mixed-precision search:</ins> Assigning quantization bit-width per layer (for weights/activations), based on the layer's sensitivity to different bit-widths.
         * <ins>Visualization:</ins> You can use TensorBoard to observe useful information for troubleshooting the quantized model's performance (for example, the model in different phases of the quantization, collected statistics, similarity between layers of the float and quantized model and bit-width configuration for mixed-precision quantization). For more details, please read the [visualization documentation](https://sony.github.io/model_optimization/docs/guidelines/visualization.html).   
+        * <ins>Target Platform Capabilities:</ins> The Target Platform Capabilities (TPC) describes the target platform (an edge device with dedicated hardware). For more details, please read the [TPC README](model_compression_toolkit/target_platform_capabilities/README.md).   
         
         
         #### Experimental features 
@@ -87,15 +88,15 @@ Description: # Model Compression Toolkit (MCT)
         pip install mct-nightly
         ```
         
-        ### Requierments
+        ### Requirements
         
-        To run MCT, one of the supported frameworks, Tenosflow/Pytorch, needs to be installed.
+        To run MCT, one of the supported frameworks, Tensorflow/Pytorch, needs to be installed.
         
-        For using with Tensorflow please install the packages: 
+        For use with Tensorflow please install the packages: 
         [tensorflow](https://www.tensorflow.org/install), 
         [tensorflow-model-optimization](https://www.tensorflow.org/model_optimization/guide/install)
         
-        For using with PyTorch please install the packages: 
+        For use with PyTorch please install the packages: 
         [torch](https://pytorch.org/)
         
         Also, a [requirements](requirements.txt) file can be used to set up your environment.

{mct-nightly-1.8.0.8032023.post421 → mct-nightly-1.8.0.8052023.post414}/mct_nightly.egg-info/SOURCES.txt

@@ -8,19 +8,18 @@ mct_nightly.egg-info/dependency_links.txt
 mct_nightly.egg-info/requires.txt
 mct_nightly.egg-info/top_level.txt
 model_compression_toolkit/__init__.py
+model_compression_toolkit/constants.py
+model_compression_toolkit/logger.py
 model_compression_toolkit/core/__init__.py
 model_compression_toolkit/core/analyzer.py
 model_compression_toolkit/core/exporter.py
 model_compression_toolkit/core/runner.py
 model_compression_toolkit/core/common/__init__.py
 model_compression_toolkit/core/common/base_substitutions.py
-model_compression_toolkit/core/common/constants.py
 model_compression_toolkit/core/common/data_loader.py
 model_compression_toolkit/core/common/defaultdict.py
 model_compression_toolkit/core/common/framework_implementation.py
 model_compression_toolkit/core/common/framework_info.py
-model_compression_toolkit/core/common/immutable.py
-model_compression_toolkit/core/common/logger.py
 model_compression_toolkit/core/common/memory_computation.py
 model_compression_toolkit/core/common/model_builder_mode.py
 model_compression_toolkit/core/common/model_collector.py
@@ -129,20 +128,6 @@ model_compression_toolkit/core/common/substitutions/shift_negative_activation.py
 model_compression_toolkit/core/common/substitutions/softmax_shift.py
 model_compression_toolkit/core/common/substitutions/virtual_activation_weights_composition.py
 model_compression_toolkit/core/common/substitutions/weights_activation_split.py
-model_compression_toolkit/core/common/target_platform/__init__.py
-model_compression_toolkit/core/common/target_platform/current_tp_model.py
-model_compression_toolkit/core/common/target_platform/fusing.py
-model_compression_toolkit/core/common/target_platform/op_quantization_config.py
-model_compression_toolkit/core/common/target_platform/operators.py
-model_compression_toolkit/core/common/target_platform/target_platform_model.py
-model_compression_toolkit/core/common/target_platform/target_platform_model_component.py
-model_compression_toolkit/core/common/target_platform/targetplatform2framework/__init__.py
-model_compression_toolkit/core/common/target_platform/targetplatform2framework/attribute_filter.py
-model_compression_toolkit/core/common/target_platform/targetplatform2framework/current_tpc.py
-model_compression_toolkit/core/common/target_platform/targetplatform2framework/layer_filter_params.py
-model_compression_toolkit/core/common/target_platform/targetplatform2framework/operations_to_layers.py
-model_compression_toolkit/core/common/target_platform/targetplatform2framework/target_platform_capabilities.py
-model_compression_toolkit/core/common/target_platform/targetplatform2framework/target_platform_capabilities_component.py
 model_compression_toolkit/core/common/visualization/__init__.py
 model_compression_toolkit/core/common/visualization/final_config_visualizer.py
 model_compression_toolkit/core/common/visualization/nn_visualizer.py
@@ -154,7 +139,6 @@ model_compression_toolkit/core/keras/keras_implementation.py
 model_compression_toolkit/core/keras/keras_model_validation.py
 model_compression_toolkit/core/keras/keras_node_prior_info.py
 model_compression_toolkit/core/keras/kpi_data_facade.py
-model_compression_toolkit/core/keras/quantization_facade.py
 model_compression_toolkit/core/keras/tf_tensor_numpy.py
 model_compression_toolkit/core/keras/back2framework/__init__.py
 model_compression_toolkit/core/keras/back2framework/factory_model_builder.py
@@ -213,7 +197,6 @@ model_compression_toolkit/core/pytorch/default_framework_info.py
 model_compression_toolkit/core/pytorch/kpi_data_facade.py
 model_compression_toolkit/core/pytorch/pytorch_implementation.py
 model_compression_toolkit/core/pytorch/pytorch_node_prior_info.py
-model_compression_toolkit/core/pytorch/quantization_facade.py
 model_compression_toolkit/core/pytorch/utils.py
 model_compression_toolkit/core/pytorch/back2framework/__init__.py
 model_compression_toolkit/core/pytorch/back2framework/factory_model_builder.py
@@ -255,60 +238,6 @@ model_compression_toolkit/core/pytorch/reader/node_holders.py
 model_compression_toolkit/core/pytorch/reader/reader.py
 model_compression_toolkit/core/pytorch/statistics_correction/__init__.py
 model_compression_toolkit/core/pytorch/statistics_correction/apply_second_moment_correction.py
-model_compression_toolkit/core/tpc_models/__init__.py
-model_compression_toolkit/core/tpc_models/get_target_platform_capabilities.py
-model_compression_toolkit/core/tpc_models/default_tpc/__init__.py
-model_compression_toolkit/core/tpc_models/default_tpc/target_platform_capabilities.py
-model_compression_toolkit/core/tpc_models/default_tpc/latest/__init__.py
-model_compression_toolkit/core/tpc_models/default_tpc/v1/__init__.py
-model_compression_toolkit/core/tpc_models/default_tpc/v1/tp_model.py
-model_compression_toolkit/core/tpc_models/default_tpc/v1/tpc_keras.py
-model_compression_toolkit/core/tpc_models/default_tpc/v1/tpc_pytorch.py
-model_compression_toolkit/core/tpc_models/default_tpc/v2/__init__.py
-model_compression_toolkit/core/tpc_models/default_tpc/v2/tp_model.py
-model_compression_toolkit/core/tpc_models/default_tpc/v2/tpc_keras.py
-model_compression_toolkit/core/tpc_models/default_tpc/v2/tpc_pytorch.py
-model_compression_toolkit/core/tpc_models/default_tpc/v3/__init__.py
-model_compression_toolkit/core/tpc_models/default_tpc/v3/tp_model.py
-model_compression_toolkit/core/tpc_models/default_tpc/v3/tpc_keras.py
-model_compression_toolkit/core/tpc_models/default_tpc/v3/tpc_pytorch.py
-model_compression_toolkit/core/tpc_models/default_tpc/v3_lut/__init__.py
-model_compression_toolkit/core/tpc_models/default_tpc/v3_lut/tp_model.py
-model_compression_toolkit/core/tpc_models/default_tpc/v3_lut/tpc_keras.py
-model_compression_toolkit/core/tpc_models/default_tpc/v3_lut/tpc_pytorch.py
-model_compression_toolkit/core/tpc_models/default_tpc/v4/__init__.py
-model_compression_toolkit/core/tpc_models/default_tpc/v4/tp_model.py
-model_compression_toolkit/core/tpc_models/default_tpc/v4/tpc_keras.py
-model_compression_toolkit/core/tpc_models/default_tpc/v4/tpc_pytorch.py
-model_compression_toolkit/core/tpc_models/default_tpc/v4_lut/__init__.py
-model_compression_toolkit/core/tpc_models/default_tpc/v4_lut/tp_model.py
-model_compression_toolkit/core/tpc_models/default_tpc/v4_lut/tpc_keras.py
-model_compression_toolkit/core/tpc_models/default_tpc/v4_lut/tpc_pytorch.py
-model_compression_toolkit/core/tpc_models/default_tpc/v5/__init__.py
-model_compression_toolkit/core/tpc_models/default_tpc/v5/tp_model.py
-model_compression_toolkit/core/tpc_models/default_tpc/v5/tpc_keras.py
-model_compression_toolkit/core/tpc_models/default_tpc/v5/tpc_pytorch.py
-model_compression_toolkit/core/tpc_models/imx500_tpc/__init__.py
-model_compression_toolkit/core/tpc_models/imx500_tpc/target_platform_capabilities.py
-model_compression_toolkit/core/tpc_models/imx500_tpc/latest/__init__.py
-model_compression_toolkit/core/tpc_models/imx500_tpc/v1/__init__.py
-model_compression_toolkit/core/tpc_models/imx500_tpc/v1/tp_model.py
-model_compression_toolkit/core/tpc_models/imx500_tpc/v1/tpc_keras.py
-model_compression_toolkit/core/tpc_models/imx500_tpc/v1/tpc_pytorch.py
-model_compression_toolkit/core/tpc_models/qnnpack_tpc/__init__.py
-model_compression_toolkit/core/tpc_models/qnnpack_tpc/target_platform_capabilities.py
-model_compression_toolkit/core/tpc_models/qnnpack_tpc/latest/__init__.py
-model_compression_toolkit/core/tpc_models/qnnpack_tpc/v1/__init__.py
-model_compression_toolkit/core/tpc_models/qnnpack_tpc/v1/tp_model.py
-model_compression_toolkit/core/tpc_models/qnnpack_tpc/v1/tpc_keras.py
-model_compression_toolkit/core/tpc_models/qnnpack_tpc/v1/tpc_pytorch.py
-model_compression_toolkit/core/tpc_models/tflite_tpc/__init__.py
-model_compression_toolkit/core/tpc_models/tflite_tpc/target_platform_capabilities.py
-model_compression_toolkit/core/tpc_models/tflite_tpc/latest/__init__.py
-model_compression_toolkit/core/tpc_models/tflite_tpc/v1/__init__.py
-model_compression_toolkit/core/tpc_models/tflite_tpc/v1/tp_model.py
-model_compression_toolkit/core/tpc_models/tflite_tpc/v1/tpc_keras.py
-model_compression_toolkit/core/tpc_models/tflite_tpc/v1/tpc_pytorch.py
 model_compression_toolkit/exporter/__init__.py
 model_compression_toolkit/exporter/model_exporter/__init__.py
 model_compression_toolkit/exporter/model_exporter/fw_agonstic/__init__.py
@@ -344,10 +273,11 @@ model_compression_toolkit/gptq/runner.py
 model_compression_toolkit/gptq/common/__init__.py
 model_compression_toolkit/gptq/common/gptq_config.py
 model_compression_toolkit/gptq/common/gptq_constants.py
+model_compression_toolkit/gptq/common/gptq_framework_implementation.py
 model_compression_toolkit/gptq/common/gptq_graph.py
-model_compression_toolkit/gptq/common/gptq_quantizer_config.py
 model_compression_toolkit/gptq/common/gptq_training.py
 model_compression_toolkit/gptq/keras/__init__.py
+model_compression_toolkit/gptq/keras/gptq_keras_implementation.py
 model_compression_toolkit/gptq/keras/gptq_loss.py
 model_compression_toolkit/gptq/keras/gptq_training.py
 model_compression_toolkit/gptq/keras/graph_info.py
@@ -356,12 +286,16 @@ model_compression_toolkit/gptq/keras/quantizer/__init__.py
 model_compression_toolkit/gptq/keras/quantizer/base_keras_gptq_quantizer.py
 model_compression_toolkit/gptq/keras/quantizer/quant_utils.py
 model_compression_toolkit/gptq/keras/quantizer/quantization_builder.py
+model_compression_toolkit/gptq/keras/quantizer/regularization_factory.py
 model_compression_toolkit/gptq/keras/quantizer/soft_rounding/__init__.py
+model_compression_toolkit/gptq/keras/quantizer/soft_rounding/soft_quantizer_reg.py
 model_compression_toolkit/gptq/keras/quantizer/soft_rounding/symmetric_soft_quantizer.py
+model_compression_toolkit/gptq/keras/quantizer/soft_rounding/uniform_soft_quantizer.py
 model_compression_toolkit/gptq/keras/quantizer/ste_rounding/__init__.py
 model_compression_toolkit/gptq/keras/quantizer/ste_rounding/symmetric_ste.py
 model_compression_toolkit/gptq/pytorch/__init__.py
 model_compression_toolkit/gptq/pytorch/gptq_loss.py
+model_compression_toolkit/gptq/pytorch/gptq_pytorch_implementation.py
 model_compression_toolkit/gptq/pytorch/gptq_training.py
 model_compression_toolkit/gptq/pytorch/graph_info.py
 model_compression_toolkit/gptq/pytorch/quantization_facade.py
@@ -369,10 +303,16 @@ model_compression_toolkit/gptq/pytorch/quantizer/__init__.py
 model_compression_toolkit/gptq/pytorch/quantizer/base_pytorch_gptq_quantizer.py
 model_compression_toolkit/gptq/pytorch/quantizer/quant_utils.py
 model_compression_toolkit/gptq/pytorch/quantizer/quantization_builder.py
+model_compression_toolkit/gptq/pytorch/quantizer/regularization_factory.py
 model_compression_toolkit/gptq/pytorch/quantizer/soft_rounding/__init__.py
+model_compression_toolkit/gptq/pytorch/quantizer/soft_rounding/soft_quantizer_reg.py
 model_compression_toolkit/gptq/pytorch/quantizer/soft_rounding/symmetric_soft_quantizer.py
+model_compression_toolkit/gptq/pytorch/quantizer/soft_rounding/uniform_soft_quantizer.py
 model_compression_toolkit/gptq/pytorch/quantizer/ste_rounding/__init__.py
 model_compression_toolkit/gptq/pytorch/quantizer/ste_rounding/symmetric_ste.py
+model_compression_toolkit/legacy/__init__.py
+model_compression_toolkit/legacy/keras_quantization_facade.py
+model_compression_toolkit/legacy/pytorch_quantization_facade.py
 model_compression_toolkit/ptq/__init__.py
 model_compression_toolkit/ptq/runner.py
 model_compression_toolkit/ptq/keras/__init__.py
@@ -381,7 +321,6 @@ model_compression_toolkit/ptq/pytorch/__init__.py
 model_compression_toolkit/ptq/pytorch/quantization_facade.py
 model_compression_toolkit/qat/__init__.py
 model_compression_toolkit/qat/common/__init__.py
-model_compression_toolkit/qat/common/constants.py
 model_compression_toolkit/qat/common/qat_config.py
 model_compression_toolkit/qat/keras/__init__.py
 model_compression_toolkit/qat/keras/quantization_facade.py
@@ -402,9 +341,9 @@ model_compression_toolkit/qat/pytorch/quantizer/ste_rounding/__init__.py
 model_compression_toolkit/qat/pytorch/quantizer/ste_rounding/symmetric_ste.py
 model_compression_toolkit/qat/pytorch/quantizer/ste_rounding/uniform_ste.py
 model_compression_toolkit/quantizers_infrastructure/__init__.py
+model_compression_toolkit/quantizers_infrastructure/constants.py
 model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/__init__.py
 model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/common/__init__.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/common/activation_lut_pot_inferable_quantizer.py
 model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/common/base_inferable_quantizer.py
 model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/common/constants.py
 model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/common/get_all_subclasses.py
@@ -439,6 +378,7 @@ model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pyt
 model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/base_uniform_inferable_quantizer.py
 model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/constants.py
 model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/activation_inferable_quantizers/__init__.py
+model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/activation_inferable_quantizers/activation_lut_pot_inferable_quantizer.py
 model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/activation_inferable_quantizers/activation_pot_inferable_quantizer.py
 model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/activation_inferable_quantizers/activation_symmetric_inferable_quantizer.py
 model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/activation_inferable_quantizers/activation_uniform_inferable_quantizer.py
@@ -460,4 +400,76 @@ model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/ker
 model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/keras/config_serialization.py
 model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/keras/quantizer_utils.py
 model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/pytorch/__init__.py
-model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/pytorch/base_pytorch_quantizer.py
+model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/pytorch/base_pytorch_quantizer.py
+model_compression_toolkit/target_platform_capabilities/__init__.py
+model_compression_toolkit/target_platform_capabilities/constants.py
+model_compression_toolkit/target_platform_capabilities/immutable.py
+model_compression_toolkit/target_platform_capabilities/target_platform/__init__.py
+model_compression_toolkit/target_platform_capabilities/target_platform/current_tp_model.py
+model_compression_toolkit/target_platform_capabilities/target_platform/fusing.py
+model_compression_toolkit/target_platform_capabilities/target_platform/op_quantization_config.py
+model_compression_toolkit/target_platform_capabilities/target_platform/operators.py
+model_compression_toolkit/target_platform_capabilities/target_platform/quantization_format.py
+model_compression_toolkit/target_platform_capabilities/target_platform/target_platform_model.py
+model_compression_toolkit/target_platform_capabilities/target_platform/target_platform_model_component.py
+model_compression_toolkit/target_platform_capabilities/target_platform/targetplatform2framework/__init__.py
+model_compression_toolkit/target_platform_capabilities/target_platform/targetplatform2framework/attribute_filter.py
+model_compression_toolkit/target_platform_capabilities/target_platform/targetplatform2framework/current_tpc.py
+model_compression_toolkit/target_platform_capabilities/target_platform/targetplatform2framework/layer_filter_params.py
+model_compression_toolkit/target_platform_capabilities/target_platform/targetplatform2framework/operations_to_layers.py
+model_compression_toolkit/target_platform_capabilities/target_platform/targetplatform2framework/target_platform_capabilities.py
+model_compression_toolkit/target_platform_capabilities/target_platform/targetplatform2framework/target_platform_capabilities_component.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/get_target_platform_capabilities.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/target_platform_capabilities.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/latest/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v1/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v1/tp_model.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v1/tpc_keras.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v1/tpc_pytorch.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v2/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v2/tp_model.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v2/tpc_keras.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v2/tpc_pytorch.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3/tp_model.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3/tpc_keras.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3/tpc_pytorch.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3_lut/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3_lut/tp_model.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3_lut/tpc_keras.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v3_lut/tpc_pytorch.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4/tp_model.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4/tpc_keras.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4/tpc_pytorch.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4_lut/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4_lut/tp_model.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4_lut/tpc_keras.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v4_lut/tpc_pytorch.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v5/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v5/tp_model.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v5/tpc_keras.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/default_tpc/v5/tpc_pytorch.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/target_platform_capabilities.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/latest/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1/tp_model.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1/tpc_keras.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/imx500_tpc/v1/tpc_pytorch.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/qnnpack_tpc/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/qnnpack_tpc/target_platform_capabilities.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/qnnpack_tpc/latest/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/qnnpack_tpc/v1/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/qnnpack_tpc/v1/tp_model.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/qnnpack_tpc/v1/tpc_keras.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/qnnpack_tpc/v1/tpc_pytorch.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/target_platform_capabilities.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/latest/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/v1/__init__.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/v1/tp_model.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/v1/tpc_keras.py
+model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/v1/tpc_pytorch.py

mct-nightly-1.8.0.8052023.post414/model_compression_toolkit/__init__.py

@@ -0,0 +1,31 @@
+# Copyright 2021 Sony Semiconductor Israel, Inc. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+
+
+from model_compression_toolkit.target_platform_capabilities import target_platform
+from model_compression_toolkit.target_platform_capabilities.tpc_models.get_target_platform_capabilities import get_target_platform_capabilities
+from model_compression_toolkit import core
+from model_compression_toolkit.logger import set_log_folder
+from model_compression_toolkit.legacy.keras_quantization_facade import keras_post_training_quantization, keras_post_training_quantization_mixed_precision
+from model_compression_toolkit.legacy.pytorch_quantization_facade import pytorch_post_training_quantization, pytorch_post_training_quantization_mixed_precision
+from model_compression_toolkit import quantizers_infrastructure
+from model_compression_toolkit import ptq
+from model_compression_toolkit import qat
+from model_compression_toolkit import exporter
+from model_compression_toolkit import gptq
+from model_compression_toolkit.gptq import GradientPTQConfig
+
+
+__version__ = "1.8.0"

{mct-nightly-1.8.0.8032023.post421/model_compression_toolkit/core/common → mct-nightly-1.8.0.8052023.post414/model_compression_toolkit}/constants.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
+
 import importlib
 
 # Supported frameworks in MCT:
@@ -47,9 +48,6 @@ LAST_AXIS = -1
 DATA_TYPE = 'dtype'
 FLOAT_32 = 'float32'
 
-# Version
-LATEST = 'latest'
-
 # Number of Tensorboard cosine-similarity plots to add:
 NUM_SAMPLES_DISTANCE_TENSORBOARD = 20
 
@@ -119,12 +117,10 @@ WEIGHTS_CHANNELS_AXIS = 'weights_channels_axis'
 DUMMY_NODE = 'dummy_node'
 DUMMY_TENSOR = 'dummy_tensor'
 
-# TP Model constants
-OPS_SET_LIST = 'ops_set_list'
 
 # TF Input node base name
 INPUT_BASE_NAME = 'base_input'
 
 # Jacobian-weights constants
 MIN_JACOBIANS_ITER = 10
-JACOBIANS_COMP_TOLERANCE = 1e-3
+JACOBIANS_COMP_TOLERANCE = 1e-3

mct-nightly-1.8.0.8052023.post414/model_compression_toolkit/core/__init__.py

@@ -0,0 +1,28 @@
+# Copyright 2022 Sony Semiconductor Israel, Inc. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+
+from model_compression_toolkit.core.common.data_loader import FolderImageLoader
+from model_compression_toolkit.core.common.framework_info import FrameworkInfo, ChannelAxis
+from model_compression_toolkit.core.common.defaultdict import DefaultDict
+from model_compression_toolkit.core.common import network_editors as network_editor
+from model_compression_toolkit.core.common.quantization.debug_config import DebugConfig
+from model_compression_toolkit.core.common.quantization import quantization_config
+from model_compression_toolkit.core.common.mixed_precision import mixed_precision_quantization_config
+from model_compression_toolkit.core.common.quantization.quantization_config import QuantizationConfig, QuantizationErrorMethod, DEFAULTCONFIG
+from model_compression_toolkit.core.common.quantization.core_config import CoreConfig
+from model_compression_toolkit.core.common.mixed_precision.kpi_tools.kpi import KPI
+from model_compression_toolkit.core.common.mixed_precision.mixed_precision_quantization_config import MixedPrecisionQuantizationConfig, MixedPrecisionQuantizationConfigV2
+from model_compression_toolkit.core.keras.kpi_data_facade import keras_kpi_data, keras_kpi_data_experimental
+from model_compression_toolkit.core.pytorch.kpi_data_facade import pytorch_kpi_data, pytorch_kpi_data_experimental

{mct-nightly-1.8.0.8032023.post421 → mct-nightly-1.8.0.8052023.post414}/model_compression_toolkit/core/analyzer.py

@@ -17,14 +17,15 @@
 from typing import Callable
 
 from model_compression_toolkit.core.common.framework_implementation import FrameworkImplementation
-from model_compression_toolkit.core.common import FrameworkInfo, Logger
-from model_compression_toolkit.core.common.constants import NUM_SAMPLES_DISTANCE_TENSORBOARD
+from model_compression_toolkit.core.common import FrameworkInfo
+from model_compression_toolkit.constants import NUM_SAMPLES_DISTANCE_TENSORBOARD
 from model_compression_toolkit.core.common.graph.base_graph import Graph
 
 from model_compression_toolkit.core.common.similarity_analyzer import compute_cs
 from model_compression_toolkit.core.common.visualization.nn_visualizer import NNVisualizer
 
 from model_compression_toolkit.core.common.visualization.tensorboard_writer import TensorboardWriter
+from model_compression_toolkit.logger import Logger
 
 
 def analyzer_model_quantization(representative_data_gen: Callable,

{mct-nightly-1.8.0.8032023.post421 → mct-nightly-1.8.0.8052023.post414}/model_compression_toolkit/core/common/__init__.py

@@ -17,7 +17,6 @@ from model_compression_toolkit.core.common.base_substitutions import BaseSubstit
 from model_compression_toolkit.core.common.framework_info import FrameworkInfo
 from model_compression_toolkit.core.common.graph.base_graph import Graph
 from model_compression_toolkit.core.common.graph.base_node import BaseNode
-from model_compression_toolkit.core.common.logger import Logger
 from model_compression_toolkit.core.common.quantization.quantization_config import QuantizationConfig, DEFAULTCONFIG
 from model_compression_toolkit.core.common.quantization.quantizers.quantizers_helpers import max_power_of_two
 from model_compression_toolkit.core.common.collectors.statistics_collector import StatsCollector, NoStatsCollector

{mct-nightly-1.8.0.8032023.post421 → mct-nightly-1.8.0.8052023.post414}/model_compression_toolkit/core/common/collectors/base_collector.py

@@ -14,7 +14,7 @@
 # ==============================================================================
 
 import numpy as np
-from model_compression_toolkit.core.common.logger import Logger
+from model_compression_toolkit.logger import Logger
 
 
 class BaseCollector(object):

{mct-nightly-1.8.0.8032023.post421 → mct-nightly-1.8.0.8052023.post414}/model_compression_toolkit/core/common/collectors/mean_collector.py

@@ -17,7 +17,7 @@
 import numpy as np
 
 from model_compression_toolkit.core.common.collectors.base_collector import BaseCollector
-from model_compression_toolkit.core.common.constants import LAST_AXIS
+from model_compression_toolkit.constants import LAST_AXIS
 
 
 class MeanCollector(BaseCollector):

{mct-nightly-1.8.0.8032023.post421 → mct-nightly-1.8.0.8052023.post414}/model_compression_toolkit/core/common/collectors/min_max_per_channel_collector.py

@@ -16,7 +16,7 @@
 import numpy as np
 
 from model_compression_toolkit.core.common.collectors.base_collector import BaseCollector
-from model_compression_toolkit.core.common.constants import LAST_AXIS
+from model_compression_toolkit.constants import LAST_AXIS
 
 
 class MinMaxPerChannelCollector(BaseCollector):