TrainingExtensions/tensorflow/src/python/aimet_tensorflow/quantsim_config/quantsim_config.py
def _set_op_type_configs(self, op_configs: OpTypeType):
        """
        Set configurations for all ops of specific types (third level of specificity in configuration file)
        :param op_configs: Dictionary containing configurations for ops of certain types
        """
        modified_quantize_ops = {}
        # Set op type configs for named modules
        for op, input_output_quantize_ops in self._op_to_quantizer_lists_dict.items():
            onnx_types = self._onnx_conn_graph_name_mapper.get_onnx_type_from_conn_graph_type(op.type)
            if not onnx_types:
                continue
            for onnx_type in onnx_types:
                if onnx_type in op_configs:
                    op_config = op_configs[onnx_type]
                    self._set_config_for_op(input_output_quantize_ops, op_config, modified_quantize_ops, op)

        Set configurations for all ops of specific types (third level of specificity in configuration file)
        :param op_configs: Dictionary containing configurations for ops of certain types
        
Set configurations for all ops of specific types (third level of specificity in configuration file)
Set configurations for all ops of specific types ( third level of specificity in configuration file )
def _set_op_type_configs(self, op_configs: OpTypeType):
        modified_quantize_ops = {}
        for op, input_output_quantize_ops in self._op_to_quantizer_lists_dict.items():
            onnx_types = self._onnx_conn_graph_name_mapper.get_onnx_type_from_conn_graph_type(op.type)
            if not onnx_types:
                continue
            for onnx_type in onnx_types:
                if onnx_type in op_configs:
                    op_config = op_configs[onnx_type]
                    self._set_config_for_op(input_output_quantize_ops, op_config, modified_quantize_ops, op)
def _set_op_type_configs ( self , op_configs : OpTypeType ) : modified_quantize_ops = { } for op , input_output_quantize_ops in self . _op_to_quantizer_lists_dict . items ( ) : onnx_types = self . _onnx_conn_graph_name_mapper . get_onnx_type_from_conn_graph_type ( op . type ) if not onnx_types : continue for onnx_type in onnx_types : if onnx_type in op_configs : op_config = op_configs [ onnx_type ] self . _set_config_for_op ( input_output_quantize_ops , op_config , modified_quantize_ops , op )
Set configurations for all ops of specific types (third level of specificity in configuration file)
Set configurations for all ops of specific types ( third level of specificity in configuration file )
"""
        Set configurations for all ops of specific types (third level of specificity in configuration file)
        :param op_configs: Dictionary containing configurations for ops of certain types
        """ # Set op type configs for named modules
self
op_configs
OpTypeType
Dictionary containing configurations for ops of certain types
Dictionary containing configurations for ops of certain types