mct-nightly 2.3.0.20250403.518__py3-none-any.whl → 2.3.0.20250405.519__py3-none-any.whl

{mct_nightly-2.3.0.20250403.518.dist-info → mct_nightly-2.3.0.20250405.519.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mct-nightly
-Version: 2.3.0.20250403.518
+Version: 2.3.0.20250405.519
 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.20250403.518.dist-info → mct_nightly-2.3.0.20250405.519.dist-info}/RECORD

@@ -1,5 +1,5 @@
-mct_nightly-2.3.0.20250403.518.dist-info/licenses/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
-model_compression_toolkit/__init__.py,sha256=z7w2tRBoJC1dhCtnjEKyK834X-V0TBq_pKLiNWkHc5s,1557
+mct_nightly-2.3.0.20250405.519.dist-info/licenses/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
+model_compression_toolkit/__init__.py,sha256=q-6vPTQu-LuGF_ciYO6xH7ip2LG_WgfTyo51-zRt4so,1557
 model_compression_toolkit/constants.py,sha256=2ltuH-gdaLZoZV4CPUgKjC3S9ojz2z4OTVdenyVEypU,3912
 model_compression_toolkit/defaultdict.py,sha256=LSc-sbZYXENMCw3U9F4GiXuv67IKpdn0Qm7Fr11jy-4,2277
 model_compression_toolkit/logger.py,sha256=L3q7tn3Uht0i_7phnlOWMR2Te2zvzrt2HOz9vYEInts,4529
@@ -435,14 +435,14 @@ model_compression_toolkit/target_platform_capabilities/constants.py,sha256=BFSgD
 model_compression_toolkit/target_platform_capabilities/immutable.py,sha256=YhROBiXEIB3TU-bAFrnL3qbAsb1yuWPBAQ_CLOJbYUU,1827
 model_compression_toolkit/target_platform_capabilities/tpc_io_handler.py,sha256=4ydTWWKv_PEOAFok2JtxFNj8rav-0IlqcXKF6lnhHNE,4157
 model_compression_toolkit/target_platform_capabilities/schema/__init__.py,sha256=pKAdbTCFM_2BrZXUtTIw0ouKotrWwUDF_hP3rPwCM2k,696
-model_compression_toolkit/target_platform_capabilities/schema/mct_current_schema.py,sha256=hf539WJ3nBGn0RnALXrKmAPnbhJ-VmWmLIa207x8b4M,541
+model_compression_toolkit/target_platform_capabilities/schema/mct_current_schema.py,sha256=PvO8eHxnb3A55gyExT5fZGnOUl3ce7BbbT5SPxCEXNo,541
 model_compression_toolkit/target_platform_capabilities/schema/schema_functions.py,sha256=vBkXxVJagm9JKB9cdm4Pvi7u_luriXUjvNn0-m8Zr0k,4653
 model_compression_toolkit/target_platform_capabilities/schema/v1.py,sha256=4CGpWENuOyjwaIMaGrFI0Act7jsSeT7m94pjrv91dxE,27516
-model_compression_toolkit/target_platform_capabilities/schema/v2.py,sha256=vUhCocA0EcjdR741Yv48W4Kr5Pq22Miebhm7F9GKb3Y,6086
+model_compression_toolkit/target_platform_capabilities/schema/v2.py,sha256=yg0ZrsaqaS69lmDvxRrz636CRARzx_eZbokTMVHNEXc,4555
 model_compression_toolkit/target_platform_capabilities/targetplatform2framework/__init__.py,sha256=XjNws3zoiJkeH4ixKqrLA5xBvpv5rq31qX7wYQjNpZM,1447
 model_compression_toolkit/target_platform_capabilities/targetplatform2framework/attach2fw.py,sha256=HJ8uc3PFfyxg-WpVXPBg4mGaox8Z9bRqtQNbRfIyAk4,3745
-model_compression_toolkit/target_platform_capabilities/targetplatform2framework/attach2keras.py,sha256=AE09QLE_QKwNqUTZbkZP9XLJStG1ECiTWmEGuXZTEsQ,7652
-model_compression_toolkit/target_platform_capabilities/targetplatform2framework/attach2pytorch.py,sha256=-zbPmzQJal-1vZiQ6vIBBBnlEOB2DTb09koA0Aj4I_I,6396
+model_compression_toolkit/target_platform_capabilities/targetplatform2framework/attach2keras.py,sha256=mxc3DBbUi-HDFgSx8Nmnyxr8SIdbx8lmtcRMsQl1BLE,7578
+model_compression_toolkit/target_platform_capabilities/targetplatform2framework/attach2pytorch.py,sha256=8spnpqxVUv8WF9-PTukOLvJAFiNi01wNowUVIDqSj5I,6321
 model_compression_toolkit/target_platform_capabilities/targetplatform2framework/attribute_filter.py,sha256=jfhszvuD2Fyy6W2KjlLzXBQKFzTqGAaDZeFVr4-ONQw,8776
 model_compression_toolkit/target_platform_capabilities/targetplatform2framework/current_tpc.py,sha256=_kFG0USYa6yzvLsi82_Vusv_KR8Hi7J1u680pPXECuo,2192
 model_compression_toolkit/target_platform_capabilities/targetplatform2framework/framework_quantization_capabilities.py,sha256=UKzckLYLdBcFAptyKnVMwpPpfRkmF0SK1Kl0g0eGjQA,9710
@@ -527,7 +527,7 @@ model_compression_toolkit/xquant/pytorch/model_analyzer.py,sha256=b93o800yVB3Z-i
 model_compression_toolkit/xquant/pytorch/pytorch_report_utils.py,sha256=UVN_S9ULHBEldBpShCOt8-soT8YTQ5oE362y96qF_FA,3950
 model_compression_toolkit/xquant/pytorch/similarity_functions.py,sha256=CERxq5K8rqaiE-DlwhZBTUd9x69dtYJlkHOPLB54vm8,2354
 model_compression_toolkit/xquant/pytorch/tensorboard_utils.py,sha256=mkoEktLFFHtEKzzFRn_jCnxjhJolK12TZ5AQeDHzUO8,9767
-mct_nightly-2.3.0.20250403.518.dist-info/METADATA,sha256=D6WPQRCnXD6lqzCblmPu_dLfulyf5bSMcbH-9mm_nNI,27098
-mct_nightly-2.3.0.20250403.518.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-mct_nightly-2.3.0.20250403.518.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
-mct_nightly-2.3.0.20250403.518.dist-info/RECORD,,
+mct_nightly-2.3.0.20250405.519.dist-info/METADATA,sha256=PHbbtoK9OgtSBmT1-3MirDEL_DgGZjTARyZ99aB_VUM,27098
+mct_nightly-2.3.0.20250405.519.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+mct_nightly-2.3.0.20250405.519.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
+mct_nightly-2.3.0.20250405.519.dist-info/RECORD,,

model_compression_toolkit/__init__.py

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

model_compression_toolkit/target_platform_capabilities/schema/mct_current_schema.py

@@ -1,4 +1,4 @@
-import model_compression_toolkit.target_platform_capabilities.schema.v2 as schema
+import model_compression_toolkit.target_platform_capabilities.schema.v1 as schema
 
 OperatorSetNames = schema.OperatorSetNames
 Signedness = schema.Signedness

model_compression_toolkit/target_platform_capabilities/schema/v2.py

@@ -30,72 +30,8 @@ from model_compression_toolkit.target_platform_capabilities.schema.v1 import (
     OperatorsSetBase,
     OperatorsSet,
     OperatorSetGroup,
-    Fusing)
-
-
-class OperatorSetNames(str, Enum):
-    CONV = "Conv"
-    DEPTHWISE_CONV = "DepthwiseConv2D"
-    CONV_TRANSPOSE = "ConvTranspose"
-    FULLY_CONNECTED = "FullyConnected"
-    CONCATENATE = "Concatenate"
-    STACK = "Stack"
-    UNSTACK = "Unstack"
-    GATHER = "Gather"
-    EXPAND = "Expend"
-    BATCH_NORM = "BatchNorm"
-    L2NORM = "L2Norm"
-    RELU = "ReLU"
-    RELU6 = "ReLU6"
-    LEAKY_RELU = "LeakyReLU"
-    ELU = "Elu"
-    HARD_TANH = "HardTanh"
-    ADD = "Add"
-    SUB = "Sub"
-    MUL = "Mul"
-    DIV = "Div"
-    MIN = "Min"
-    MAX = "Max"
-    PRELU = "PReLU"
-    ADD_BIAS = "AddBias"
-    SWISH = "Swish"
-    SIGMOID = "Sigmoid"
-    SOFTMAX = "Softmax"
-    LOG_SOFTMAX = "LogSoftmax"
-    TANH = "Tanh"
-    GELU = "Gelu"
-    HARDSIGMOID = "HardSigmoid"
-    HARDSWISH = "HardSwish"
-    FLATTEN = "Flatten"
-    GET_ITEM = "GetItem"
-    RESHAPE = "Reshape"
-    UNSQUEEZE = "Unsqueeze"
-    SQUEEZE = "Squeeze"
-    PERMUTE = "Permute"
-    TRANSPOSE = "Transpose"
-    DROPOUT = "Dropout"
-    SPLIT_CHUNK = "SplitChunk"
-    MAXPOOL = "MaxPool"
-    AVGPOOL = "AvgPool"
-    SIZE = "Size"
-    SHAPE = "Shape"
-    EQUAL = "Equal"
-    ARGMAX = "ArgMax"
-    TOPK = "TopK"
-    FAKE_QUANT = "FakeQuant"
-    COMBINED_NON_MAX_SUPPRESSION = "CombinedNonMaxSuppression"
-    BOX_DECODE = "BoxDecode"
-    ZERO_PADDING2D = "ZeroPadding2D"
-    CAST = "Cast"
-    RESIZE = "Resize"
-    PAD = "Pad"
-    FOLD = "Fold"
-    STRIDED_SLICE = "StridedSlice"
-    SSD_POST_PROCESS = "SSDPostProcess"
-
-    @classmethod
-    def get_values(cls):
-        return [v.value for v in cls]
+    Fusing,
+    OperatorSetNames)
 
 
 class TargetPlatformCapabilities(BaseModel):

model_compression_toolkit/target_platform_capabilities/targetplatform2framework/attach2keras.py

@@ -93,7 +93,6 @@ class AttachTpcToKeras(AttachTpcToFramework):
             OperatorSetNames.TOPK: [tf.nn.top_k],
             OperatorSetNames.FAKE_QUANT: [tf.quantization.fake_quant_with_min_max_vars],
             OperatorSetNames.COMBINED_NON_MAX_SUPPRESSION: [tf.image.combined_non_max_suppression],
-            OperatorSetNames.BOX_DECODE: [],  # no such operator in keras
             OperatorSetNames.ZERO_PADDING2D: [ZeroPadding2D],
             OperatorSetNames.CAST: [tf.cast],
             OperatorSetNames.STRIDED_SLICE: [tf.strided_slice],

model_compression_toolkit/target_platform_capabilities/targetplatform2framework/attach2pytorch.py

@@ -98,7 +98,6 @@ class AttachTpcToPytorch(AttachTpcToFramework):
                                                         Eq('p', 2) | Eq('p', None))],
             OperatorSetNames.SSD_POST_PROCESS: [],  # no such operator in pytorch
             OperatorSetNames.COMBINED_NON_MAX_SUPPRESSION: [],  # no such operator in pytorch
-            OperatorSetNames.BOX_DECODE: []  # no such operator in pytorch
         }
 
         pytorch_linear_attr_mapping = {KERNEL_ATTR: DefaultDict(default_value=PYTORCH_KERNEL),