mct-nightly 2.3.0.20250304.517__tar.gz → 2.3.0.20250305.525__tar.gz

{mct_nightly-2.3.0.20250304.517 → mct_nightly-2.3.0.20250305.525}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: mct-nightly
-Version: 2.3.0.20250304.517
+Version: 2.3.0.20250305.525
 Summary: A Model Compression Toolkit for neural networks
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: Apache Software License

{mct_nightly-2.3.0.20250304.517 → mct_nightly-2.3.0.20250305.525}/mct_nightly.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: mct-nightly
-Version: 2.3.0.20250304.517
+Version: 2.3.0.20250305.525
 Summary: A Model Compression Toolkit for neural networks
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: Apache Software License

{mct_nightly-2.3.0.20250304.517 → mct_nightly-2.3.0.20250305.525}/model_compression_toolkit/__init__.py

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

{mct_nightly-2.3.0.20250304.517 → mct_nightly-2.3.0.20250305.525}/model_compression_toolkit/target_platform_capabilities/targetplatform2framework/attach2pytorch.py

@@ -70,7 +70,7 @@ class AttachTpcToPytorch(AttachTpcToFramework):
             OperatorSetNames.HARDSWISH: [Hardswish, hardswish],
             OperatorSetNames.FLATTEN: [Flatten, flatten],
             OperatorSetNames.GET_ITEM: [operator.getitem],
-            OperatorSetNames.RESHAPE: [reshape],
+            OperatorSetNames.RESHAPE: [reshape, torch.Tensor.view],
             OperatorSetNames.UNSQUEEZE: [unsqueeze],
             OperatorSetNames.SQUEEZE: [squeeze],
             OperatorSetNames.PERMUTE: [permute],

{mct_nightly-2.3.0.20250304.517 → mct_nightly-2.3.0.20250305.525}/model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/v1/tpc.py

@@ -137,6 +137,7 @@ def generate_tpc(default_config: OpQuantizationConfig,
     # If the QuantizationConfigOptions contains only one configuration,
     # this configuration will be used for the operation quantization:
     default_configuration_options = schema.QuantizationConfigOptions(quantization_configurations=tuple([default_config]))
+    base_configuration_options = schema.QuantizationConfigOptions(quantization_configurations=tuple([base_config]), base_config=base_config)
 
     # In TFLite, the quantized operator specifications constraint operators quantization
     # differently. For more details:
@@ -179,13 +180,13 @@ def generate_tpc(default_config: OpQuantizationConfig,
                                qc_options=default_configuration_options.clone_and_edit(
                                    fixed_zero_point=-128, fixed_scale=1 / 256))
     fc = schema.OperatorsSet(name=schema.OperatorSetNames.FULLY_CONNECTED,
-                             qc_options=default_configuration_options.clone_and_edit_weight_attribute(
+                             qc_options=base_configuration_options.clone_and_edit_weight_attribute(
                                  weights_per_channel_threshold=False))
     squeeze = schema.OperatorsSet(name=schema.OperatorSetNames.SQUEEZE,
                                   qc_options=default_configuration_options.clone_and_edit(
                                       quantization_preserving=True))
 
-    conv2d = schema.OperatorsSet(name=schema.OperatorSetNames.CONV)
+    conv2d = schema.OperatorsSet(name=schema.OperatorSetNames.CONV, qc_options=base_configuration_options)
     relu = schema.OperatorsSet(name=schema.OperatorSetNames.RELU)
     relu6 = schema.OperatorsSet(name=schema.OperatorSetNames.RELU6)
     elu = schema.OperatorsSet(name=schema.OperatorSetNames.ELU)