mct-nightly 1.8.0.20052023.post401__tar.gz → 1.8.0.20230610.post356__tar.gz

{mct-nightly-1.8.0.20052023.post401 → mct-nightly-1.8.0.20230610.post356}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mct-nightly
-Version: 1.8.0.20052023.post401
+Version: 1.8.0.20230610.post356
 Summary: A Model Compression Toolkit for neural networks
 Home-page: UNKNOWN
 License: UNKNOWN
@@ -69,15 +69,17 @@ Description: # Model Compression Toolkit (MCT)
         
         This section provides a quick starting guide. We begin with installation via source code or pip server. Then, we provide a short usage example.
         
-        ### Installation
-        See the MCT install guide for the pip package, and build from the source.
-        
-        
-        #### From Source
+        ### Setting up work environment
+        Clone the repository and install the required packages (via [requirements](requirements.txt)).
         ```
         git clone https://github.com/sony/model_optimization.git
-        python setup.py install
+        cd model_optimization
+        pip install -r requirements.txt
         ```
+        
+        ### Installation
+        See the MCT install guide for the pip package.
+        
         #### From PyPi - latest stable release
         ```
         pip install model-compression-toolkit

{mct-nightly-1.8.0.20052023.post401 → mct-nightly-1.8.0.20230610.post356}/README.md

@@ -63,15 +63,17 @@ For more details, we highly recommend visiting our project website where experim
 
 This section provides a quick starting guide. We begin with installation via source code or pip server. Then, we provide a short usage example.
 
-### Installation
-See the MCT install guide for the pip package, and build from the source.
-
-
-#### From Source
+### Setting up work environment
+Clone the repository and install the required packages (via [requirements](requirements.txt)).
 ```
 git clone https://github.com/sony/model_optimization.git
-python setup.py install
+cd model_optimization
+pip install -r requirements.txt
 ```
+
+### Installation
+See the MCT install guide for the pip package.
+
 #### From PyPi - latest stable release
 ```
 pip install model-compression-toolkit

{mct-nightly-1.8.0.20052023.post401 → mct-nightly-1.8.0.20230610.post356}/mct_nightly.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mct-nightly
-Version: 1.8.0.20052023.post401
+Version: 1.8.0.20230610.post356
 Summary: A Model Compression Toolkit for neural networks
 Home-page: UNKNOWN
 License: UNKNOWN
@@ -69,15 +69,17 @@ Description: # Model Compression Toolkit (MCT)
         
         This section provides a quick starting guide. We begin with installation via source code or pip server. Then, we provide a short usage example.
         
-        ### Installation
-        See the MCT install guide for the pip package, and build from the source.
-        
-        
-        #### From Source
+        ### Setting up work environment
+        Clone the repository and install the required packages (via [requirements](requirements.txt)).
         ```
         git clone https://github.com/sony/model_optimization.git
-        python setup.py install
+        cd model_optimization
+        pip install -r requirements.txt
         ```
+        
+        ### Installation
+        See the MCT install guide for the pip package.
+        
         #### From PyPi - latest stable release
         ```
         pip install model-compression-toolkit

{mct-nightly-1.8.0.20052023.post401 → mct-nightly-1.8.0.20230610.post356}/mct_nightly.egg-info/SOURCES.txt

@@ -340,68 +340,6 @@ model_compression_toolkit/qat/pytorch/quantizer/quantizer_utils.py
 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
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/common/constants.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/common/get_all_subclasses.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/common/get_quantizers.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/common/quant_utils.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/__init__.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/activation_quantization_holder.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/load_model.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantize_wrapper.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizer_utils.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/validation_functions.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/__init__.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/base_keras_inferable_quantizer.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/constants.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/activation_inferable_quantizers/__init__.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/activation_inferable_quantizers/activation_lut_pot_inferable_quantizer.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/activation_inferable_quantizers/activation_pot_inferable_quantizer.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/activation_inferable_quantizers/activation_symmetric_inferable_quantizer.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/activation_inferable_quantizers/activation_uniform_inferable_quantizer.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/weights_inferable_quantizers/__init__.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/weights_inferable_quantizers/weights_lut_pot_inferable_quantizer.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/weights_inferable_quantizers/weights_lut_symmetric_inferable_quantizer.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/weights_inferable_quantizers/weights_pot_inferable_quantizer.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/weights_inferable_quantizers/weights_symmetric_inferable_quantizer.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/keras/quantizers/weights_inferable_quantizers/weights_uniform_inferable_quantizer.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/__init__.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantize_wrapper.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizer_utils.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/__init__.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/base_lut_symmetric_inferable_quantizer.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/base_pytorch_inferable_quantizer.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/base_symmetric_inferable_quantizer.py
-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
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/weights_inferable_quantizers/__init__.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/weights_inferable_quantizers/weights_lut_pot_inferable_quantizer.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/weights_inferable_quantizers/weights_lut_symmetric_inferable_quantizer.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/weights_inferable_quantizers/weights_pot_inferable_quantizer.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/weights_inferable_quantizers/weights_symmetric_inferable_quantizer.py
-model_compression_toolkit/quantizers_infrastructure/inferable_infrastructure/pytorch/quantizers/weights_inferable_quantizers/weights_uniform_inferable_quantizer.py
-model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/__init__.py
-model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/common/__init__.py
-model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/common/base_trainable_quantizer.py
-model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/common/get_quantizer_config.py
-model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/common/get_quantizers.py
-model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/common/quant_utils.py
-model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/common/trainable_quantizer_config.py
-model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/keras/__init__.py
-model_compression_toolkit/quantizers_infrastructure/trainable_infrastructure/keras/base_keras_quantizer.py
-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/target_platform_capabilities/__init__.py
 model_compression_toolkit/target_platform_capabilities/constants.py
 model_compression_toolkit/target_platform_capabilities/immutable.py
@@ -473,4 +411,19 @@ model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/lat
 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
+model_compression_toolkit/target_platform_capabilities/tpc_models/tflite_tpc/v1/tpc_pytorch.py
+model_compression_toolkit/trainable_infrastructure/__init__.py
+model_compression_toolkit/trainable_infrastructure/common/__init__.py
+model_compression_toolkit/trainable_infrastructure/common/base_trainable_quantizer.py
+model_compression_toolkit/trainable_infrastructure/common/constants.py
+model_compression_toolkit/trainable_infrastructure/common/get_quantizer_config.py
+model_compression_toolkit/trainable_infrastructure/common/get_quantizers.py
+model_compression_toolkit/trainable_infrastructure/common/quant_utils.py
+model_compression_toolkit/trainable_infrastructure/common/trainable_quantizer_config.py
+model_compression_toolkit/trainable_infrastructure/keras/__init__.py
+model_compression_toolkit/trainable_infrastructure/keras/base_keras_quantizer.py
+model_compression_toolkit/trainable_infrastructure/keras/config_serialization.py
+model_compression_toolkit/trainable_infrastructure/keras/load_model.py
+model_compression_toolkit/trainable_infrastructure/keras/quantizer_utils.py
+model_compression_toolkit/trainable_infrastructure/pytorch/__init__.py
+model_compression_toolkit/trainable_infrastructure/pytorch/base_pytorch_quantizer.py

{mct-nightly-1.8.0.20052023.post401 → mct-nightly-1.8.0.20230610.post356}/mct_nightly.egg-info/requires.txt

@@ -10,3 +10,4 @@ PuLP
 matplotlib
 scipy
 protobuf
+mct-quantizers-nightly

mct-nightly-1.8.0.20230610.post356/model_compression_toolkit/__init__.py

@@ -0,0 +1,51 @@
+# 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 trainable_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.trainable_infrastructure.keras.load_model import keras_load_quantized_model
+
+
+# Old API (will not be accessible in future releases)
+from model_compression_toolkit.core.common import network_editors as network_editor
+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.debug_config import DebugConfig
+from model_compression_toolkit.core.common.quantization.quantization_config import QuantizationConfig, QuantizationErrorMethod, DEFAULTCONFIG
+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
+from model_compression_toolkit.logger import set_log_folder
+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.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.core.keras.kpi_data_facade import keras_kpi_data
+from model_compression_toolkit.core.pytorch.kpi_data_facade import pytorch_kpi_data
+from model_compression_toolkit.gptq.common.gptq_config import GradientPTQConfig
+from model_compression_toolkit.gptq.common.gptq_config import RoundingType
+from model_compression_toolkit.gptq.keras.quantization_facade import get_keras_gptq_config
+from model_compression_toolkit.gptq.pytorch.quantization_facade import get_pytorch_gptq_config
+
+__version__ = "1.8.0"

{mct-nightly-1.8.0.20052023.post401 → mct-nightly-1.8.0.20230610.post356}/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.target_platform_capabilities.target_platform.op_quantization_config import QuantizationMethod
+from model_compression_toolkit.target_platform_capabilities.target_platform import QuantizationMethod
 
 
 class ChannelAxis(Enum):

{mct-nightly-1.8.0.20052023.post401 → mct-nightly-1.8.0.20230610.post356}/model_compression_toolkit/core/keras/back2framework/instance_builder.py

@@ -31,14 +31,22 @@ class OperationHandler:
     Class to handle conversions from graph nodes to Keras operators and retrieving them.
     """
 
-    def __init__(self, graph: Graph):
+    def __init__(self, graph: Graph, wrapper: Callable = None):
+        """
+
+        Args:
+            graph: Graph to build its layers based on its nodes.
+            wrapper: Wrapper to use for wrapping the layers.
+        """
+
         # hold nodes after sorting them
         self.node_sort = list(topological_sort(graph))
 
         self.layer_to_node_dict = {}
 
         # hold dictionary from node to its equivalent Keras layer
-        self.node_to_fw_op_dict = instance_builder(self.node_sort)
+        self.node_to_fw_op_dict = instance_builder(self.node_sort,
+                                                   wrapper)
 
     def get_node_op_function(self, n: BaseNode) -> Layer:
         """
@@ -74,12 +82,7 @@ def node_builder(n: common.BaseNode) -> Layer:
         Keras layer that was built from the node.
     """
     framework_attr = copy.copy(n.framework_attr)
-    if n.layer_class is InputLayer:
-        # replace input node with identity, so can wrap it with QuantizationWrapper
-        _layer_class = Layer  # Identity
-        framework_attr = {}
-    else:
-        _layer_class = n.layer_class
+    _layer_class = n.layer_class
     framework_attr[LAYER_NAME] = n.name  # Overwrite framework name to identical graph node name
     node_instance = _layer_class.from_config(framework_attr)  # Build layer from node's configuration.
     with tf.name_scope(n.name):
@@ -90,13 +93,15 @@ def node_builder(n: common.BaseNode) -> Layer:
     return node_instance
 
 
-def instance_builder(toposort: List[BaseNode]) -> Dict[BaseNode, Layer]:
+def instance_builder(toposort: List[BaseNode],
+                     wrapper: Callable = None) -> Dict[BaseNode, Layer]:
     """
     Build a dictionary of nodes to their corresponding Keras
     layers, given a list of nodes.
 
     Args:
         toposort: List of nodes sorted topological to build their layers.
+        wrapper: Wrapper to use for wrapping the layers.
 
     Returns:
         A dictionary of nodes to their corresponding Keras layers.
@@ -106,6 +111,8 @@ def instance_builder(toposort: List[BaseNode]) -> Dict[BaseNode, Layer]:
     for n in toposort:
         if not n.reuse:  # Hold a single node in dictionary for all reused nodes from the same layer.
             keras_node = node_builder(n)
+            if wrapper is not None:
+                keras_node = wrapper(n, keras_node)
             nodes_dict.update({n: keras_node})
 
     return nodes_dict

{mct-nightly-1.8.0.20052023.post401 → mct-nightly-1.8.0.20230610.post356}/model_compression_toolkit/core/keras/back2framework/keras_model_builder.py

@@ -21,7 +21,7 @@ from packaging import version
 from model_compression_toolkit.constants import INPUT_BASE_NAME
 from model_compression_toolkit.core.common.back2framework.base_model_builder import BaseModelBuilder
 from model_compression_toolkit.core.common.user_info import UserInformation
-from model_compression_toolkit.quantizers_infrastructure.inferable_infrastructure.keras.activation_quantization_holder import ActivationQuantizationHolder
+from mct_quantizers import KerasActivationQuantizationHolder
 
 # As from Tensorflow 2.6, keras is a separate package and some classes should be imported differently.
 if version.parse(tf.__version__) < version.parse("2.6"):
@@ -113,7 +113,8 @@ class KerasModelBuilder(BaseModelBuilder):
                          return_float_outputs)
 
         # Build an OperationHandler to handle conversions from graph nodes to Keras operators.
-        self.oh = OperationHandler(self.graph)
+        self.oh = OperationHandler(self.graph,
+                                   wrapper=wrapper)
         self.wrapper = wrapper
         self.get_activation_quantizer_holder = get_activation_quantizer_holder_fn
 
@@ -121,7 +122,7 @@ class KerasModelBuilder(BaseModelBuilder):
     def use_activation_holder_during_model_building(self) -> bool:
         """
 
-        Returns: Whether the model builder uses ActivationQuantizationHolder during
+        Returns: Whether the model builder uses KerasActivationQuantizationHolder during
         model building (by adding it as a layer when converting the graph to the Keras model)
         or not. If so - the model builder expects the activation quantizers to not be wrapped
         in KerasQuantizeWrapper that was received in its init.
@@ -172,11 +173,6 @@ class KerasModelBuilder(BaseModelBuilder):
                                         for
                                         inode in self.graph.get_inputs()}
 
-        # Support adding Layer after input layers require us to store it in layer_to_node_dict
-        # dict offline (unlike other layers which stored during running).
-        for node, layer in self.oh.node_to_fw_op_dict.items():
-            if node.type == InputLayer:
-                self.oh.layer_to_node_dict[layer] = node
 
         # Build a list of the model's input tensors. Switching from a dictionary to a list
         # to keep the tensors input order, since inputs in Graph are ordered by their indices.
@@ -186,7 +182,8 @@ class KerasModelBuilder(BaseModelBuilder):
 
         # Build a dictionary from node to its output tensors, by applying the layers sequentially.
         for n in self.oh.node_sort:
-            op_func = self.oh.get_node_op_function(n)  # Get node operation function
+            op_func = self.oh.get_node_op_function(n) # Get node operation function
+
             input_tensors = self._build_input_tensors_list(n,
                                                            node_to_output_tensors_dict)  # Fetch Node inputs
             out_tensors_of_n, out_tensors_of_n_float = self._run_operation(n,  # Run node operation and fetch outputs
@@ -223,22 +220,6 @@ class KerasModelBuilder(BaseModelBuilder):
         # Build the model.
         model = tf.keras.Model(inputs=inputs_list, outputs=model_output_tensors)
 
-        if self.wrapper is not None:
-            def _wrap(layer):
-                _node = self.oh.layer_to_node_dict.get(layer)
-                if _node is not None:
-                    return self.wrapper(_node,
-                                        layer)
-
-                elif is_layer_fake_quant(layer) or isinstance(layer, ActivationQuantizationHolder):
-                    return layer
-
-                raise Exception(  # pragma: no cover
-                    f'Mismatch between keras model and graph cant find node named: '
-                    f'{get_node_name_from_layer(layer)}')
-
-            model = clone_model(model, clone_function=_wrap)
-
         return model, self.graph.user_info
 
     def _convert_node2name(self, in_node_to_output_tensors_dict):
@@ -295,8 +276,7 @@ class KerasModelBuilder(BaseModelBuilder):
         if len(input_tensors) == 0:  # Placeholder handling
             out_tensors_of_n_float = input_nodes_to_input_tensors[n]
             out_tensors_of_n = self._run_operation_activation_quantization(n,
-                                                                           out_tensors_of_n_float,
-                                                                           op_func)
+                                                                           out_tensors_of_n_float)
         else:
             input_tensors = [tensor for tensor_list in input_tensors for tensor in tensor_list]  # flat list of lists
             # Build a functional node using its args
@@ -331,24 +311,18 @@ class KerasModelBuilder(BaseModelBuilder):
 
     def _run_operation_activation_quantization(self,
                                                node: BaseNode,
-                                               node_outputs: List[TFReference],
-                                               identity_layer: Layer = None):
+                                               node_outputs: List[TFReference]):
         """
         Quantize node's activations
 
         Args:
             node: Node to quantize its activations
             node_outputs: Output tensors of the float node.
-            identity_layer: Identity layer (should be passed only when quantizing input layers)
 
         Returns:
             Quantized node's outputs.
         """
         if self.wrapper is not None:
-            # If identity layer was passed, use it for inference
-            # (needed since wrapping an Input layer can not be wrapped)
-            if identity_layer is not None:
-                node_outputs = identity_layer(node_outputs)
 
             # In case the activation quantizer is attached out of the wrapper we use get_activation_quantizer_holder
             # for the activation quantization holder (if the node's activations are quantized)

{mct-nightly-1.8.0.20052023.post401 → mct-nightly-1.8.0.20230610.post356}/model_compression_toolkit/core/pytorch/back2framework/instance_builder.py

@@ -38,12 +38,16 @@ def node_builder(n: BaseNode) -> Module:
     return node_instance
 
 
-def identity_wrapper(node: BaseNode, module: Module):
+# todo: remove. It is not used anymore
+def identity_wrapper(node: BaseNode,
+                     module: Module,
+                     include_activation_quantizers: bool):
     """
     A function which takes a computational graph node and a pytorch module and return an identity wrapping which return the layer itself
     Args:
         node: A node of mct graph.
         layer: A pytorch module
+        include_activation_quantizers: bool flag.
     Returns: pytorch module
     """
     return module