mct-nightly 1.8.0.8042023.post345__tar.gz → 1.8.0.8052023.post414__tar.gz

{mct-nightly-1.8.0.8042023.post345 → mct-nightly-1.8.0.8052023.post414}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mct-nightly
-Version: 1.8.0.8042023.post345
+Version: 1.8.0.8052023.post414
 Summary: A Model Compression Toolkit for neural networks
 Home-page: UNKNOWN
 License: UNKNOWN
@@ -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 

{mct-nightly-1.8.0.8042023.post345 → mct-nightly-1.8.0.8052023.post414}/README.md

@@ -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 

{mct-nightly-1.8.0.8042023.post345 → 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.8042023.post345
+Version: 1.8.0.8052023.post414
 Summary: A Model Compression Toolkit for neural networks
 Home-page: UNKNOWN
 License: UNKNOWN
@@ -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 

{mct-nightly-1.8.0.8042023.post345 → 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,9 +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_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
@@ -359,10 +290,12 @@ 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
@@ -377,6 +310,9 @@ model_compression_toolkit/gptq/pytorch/quantizer/soft_rounding/symmetric_soft_qu
 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
@@ -385,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
@@ -406,6 +341,7 @@ 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/base_inferable_quantizer.py
@@ -464,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.8042023.post345/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.8042023.post345 → 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.8042023.post345 → 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.8042023.post345 → 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.8042023.post345 → 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.8042023.post345 → 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):

{mct-nightly-1.8.0.8042023.post345 → mct-nightly-1.8.0.8052023.post414}/model_compression_toolkit/core/common/framework_implementation.py

@@ -17,7 +17,7 @@ from typing import Callable, Any, List, Tuple, Dict
 
 import numpy as np
 
-from model_compression_toolkit import MixedPrecisionQuantizationConfigV2
+from model_compression_toolkit.core import MixedPrecisionQuantizationConfigV2
 from model_compression_toolkit.core import common
 from model_compression_toolkit.core.common import BaseNode
 from model_compression_toolkit.core.common.collectors.statistics_collector import BaseStatsCollector
@@ -288,13 +288,6 @@ class FrameworkImplementation(ABC):
                              f'framework\'s get_substitutions_after_second_moment_correction '
                              f'method.')  # pragma: no cover
 
-    @abstractmethod
-    def get_gptq_trainer_obj(self):
-        """
-        Returns: GPTQTrainer object
-        """
-        raise NotImplemented(f'{self.__class__.__name__} have to implement the '
-                             f'framework\'s get_gptq_trainer method.')  # pragma: no cover
 
     @abstractmethod
     def get_sensitivity_evaluator(self,

{mct-nightly-1.8.0.8042023.post345 → mct-nightly-1.8.0.8052023.post414}/model_compression_toolkit/core/common/framework_info.py

@@ -22,7 +22,7 @@ from typing import Dict, Any, List
 
 from model_compression_toolkit.core.common.defaultdict import DefaultDict
 from model_compression_toolkit.core.common.graph.base_node import BaseNode
-from model_compression_toolkit.core.common.target_platform.op_quantization_config import QuantizationMethod
+from model_compression_toolkit.target_platform_capabilities.target_platform.op_quantization_config import QuantizationMethod
 
 
 class ChannelAxis(Enum):

{mct-nightly-1.8.0.8042023.post345 → mct-nightly-1.8.0.8052023.post414}/model_compression_toolkit/core/common/fusion/layer_fusing.py

@@ -16,8 +16,8 @@ import copy
 from typing import Any, List
 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.target_platform.targetplatform2framework import TargetPlatformCapabilities
-from model_compression_toolkit.core.common.target_platform.targetplatform2framework.layer_filter_params import LayerFilterParams
+from model_compression_toolkit.target_platform_capabilities.target_platform.targetplatform2framework import TargetPlatformCapabilities
+from model_compression_toolkit.target_platform_capabilities.target_platform.targetplatform2framework.layer_filter_params import LayerFilterParams
 
 
 def filter_fusing_patterns(fusing_patterns: List[List[Any]], node: BaseNode, idx: int = 0) -> List[List[Any]]:
@@ -33,7 +33,7 @@ def filter_fusing_patterns(fusing_patterns: List[List[Any]], node: BaseNode, idx
     valid_fusing_patterns = []
     for i,fusing_pattern in enumerate(fusing_patterns):
         if idx < len(fusing_pattern):
-            if (type(fusing_pattern[idx]) == LayerFilterParams and fusing_pattern[idx].match(node)) or fusing_pattern[idx] == node.type:
+            if (type(fusing_pattern[idx]) == LayerFilterParams and node.is_match_filter_params(fusing_pattern[idx])) or fusing_pattern[idx] == node.type:
                 valid_fusing_patterns.append(fusing_pattern)
 
     # Return only valid patterns for this node
@@ -57,7 +57,7 @@ def is_valid_fusion(fusing_patterns: List[List[Any]], nodes: List[BaseNode]) ->
             continue
         counter = 0
         for i,layer in enumerate(fusing_pattern):
-            if (type(layer) == LayerFilterParams and layer.match(nodes[i])) or layer == nodes[i].type:
+            if (type(layer) == LayerFilterParams and nodes[i].is_match_filter_params(layer)) or layer == nodes[i].type:
                 counter += 1
         if counter == fusion_depth:
             return True

{mct-nightly-1.8.0.8042023.post345 → mct-nightly-1.8.0.8052023.post414}/model_compression_toolkit/core/common/graph/base_graph.py

@@ -30,8 +30,8 @@ from model_compression_toolkit.core.common.graph.base_node import BaseNode
 from model_compression_toolkit.core.common.collectors.statistics_collector import BaseStatsCollector
 from model_compression_toolkit.core.common.collectors.statistics_collector import scale_statistics, shift_statistics
 from model_compression_toolkit.core.common.user_info import UserInformation
-from model_compression_toolkit.core.common.logger import Logger
-from model_compression_toolkit.core.common.target_platform.targetplatform2framework import TargetPlatformCapabilities
+from model_compression_toolkit.logger import Logger
+from model_compression_toolkit.target_platform_capabilities.target_platform.targetplatform2framework import TargetPlatformCapabilities
 
 OutTensor = namedtuple('OutTensor', 'node node_out_index')
 

{mct-nightly-1.8.0.8042023.post345 → mct-nightly-1.8.0.8052023.post414}/model_compression_toolkit/core/common/graph/base_node.py

@@ -18,8 +18,11 @@ from typing import Dict, Any, Tuple, List
 
 import numpy as np
 
-from model_compression_toolkit.core.common.constants import WEIGHTS_NBITS_ATTRIBUTE, CORRECTED_BIAS_ATTRIBUTE, \
+from model_compression_toolkit.constants import WEIGHTS_NBITS_ATTRIBUTE, CORRECTED_BIAS_ATTRIBUTE, \
     ACTIVATION_NBITS_ATTRIBUTE
+from model_compression_toolkit.logger import Logger
+from model_compression_toolkit.target_platform_capabilities.target_platform import QuantizationConfigOptions, \
+    TargetPlatformCapabilities, LayerFilterParams
 
 
 class BaseNode:
@@ -429,3 +432,56 @@ class BaseNode:
 
         return len(self.candidates_quantization_cfg) > 0 and \
                any([c.activation_quantization_cfg.enable_activation_quantization for c in self.candidates_quantization_cfg])
+
+    def get_qco(self, tpc: TargetPlatformCapabilities) -> QuantizationConfigOptions:
+        """
+        Get the QuantizationConfigOptions of the node according
+        to the mappings from layers/LayerFilterParams to the OperatorsSet in the TargetPlatformModel.
+
+        Args:
+            tpc: TPC to extract the QuantizationConfigOptions for the node
+
+        Returns:
+            QuantizationConfigOptions of the node.
+        """
+
+        if tpc is None:
+            Logger.error(f'Can not retrieve QC options for None TPC')  # pragma: no cover
+
+        for fl, qco in tpc.filterlayer2qco.items():
+            if self.is_match_filter_params(fl):
+                return qco
+        if self.type in tpc.layer2qco:
+            return tpc.layer2qco.get(self.type)
+        return tpc.tp_model.default_qco
+
+
+    def is_match_filter_params(self, layer_filter_params: LayerFilterParams) -> bool:
+        """
+        Check if the node matches a LayerFilterParams according to its
+        layer, conditions and keyword-arguments.
+
+        Args:
+            layer_filter_params: LayerFilterParams to check if the node matches its properties.
+
+        Returns:
+            Whether the node matches to the LayerFilterParams properties.
+        """
+        # Check the node has the same type as the layer in LayerFilterParams
+        if layer_filter_params.layer != self.type:
+            return False
+
+        # Get attributes from node to filter
+        layer_config = self.framework_attr
+        if hasattr(self, "op_call_kwargs"):
+            layer_config.update(self.op_call_kwargs)
+
+        for attr, value in layer_filter_params.kwargs.items():
+            if layer_config.get(attr) != value:
+                return False
+
+        for c in layer_filter_params.conditions:
+            if not c.match(layer_config):
+                return False
+
+        return True