TrainingExtensions/tensorflow/src/python/aimet_tensorflow/quantsim_config/quantsim_config.py
def _set_default_configs_for_ops(self, default_op_configs: ConfigType):
        """
        Set default configurations for all ops in the model.
        :param default_op_configs: Default configurations for ops
        """
        # Modified quantize ops is a dictionary that keeps track of the quantize ops that are modified by this function.
        # It is initialized as empty, and each time self._set_config_for_op is called, an op will be added.
        # If ever a quantize op's attribute has already been modified this round and a contradicting setting is
        # specified, an assertion will be thrown.
        modified_quantize_ops = {}
        input_ops = get_all_input_ops(self._conn_graph)
        for op, input_output_quantize_ops in self._op_to_quantizer_lists_dict.items():
            if op in input_ops:
                # Don't set the input op setting, since it should be off by default even if all ops within the model
                # are to be quantized.  If model input is to be quantized, model_input should be set in config file
                # instead.
                continue
            self._set_config_for_op(input_output_quantize_ops, default_op_configs, modified_quantize_ops)

        Set default configurations for all ops in the model.
        :param default_op_configs: Default configurations for ops
        
Set default configurations for all ops in the model.
Set default configurations for all ops in the model .
def _set_default_configs_for_ops(self, default_op_configs: ConfigType):
        modified_quantize_ops = {}
        input_ops = get_all_input_ops(self._conn_graph)
        for op, input_output_quantize_ops in self._op_to_quantizer_lists_dict.items():
            if op in input_ops:
                continue
            self._set_config_for_op(input_output_quantize_ops, default_op_configs, modified_quantize_ops)
def _set_default_configs_for_ops ( self , default_op_configs : ConfigType ) : modified_quantize_ops = { } input_ops = get_all_input_ops ( self . _conn_graph ) for op , input_output_quantize_ops in self . _op_to_quantizer_lists_dict . items ( ) : if op in input_ops : continue self . _set_config_for_op ( input_output_quantize_ops , default_op_configs , modified_quantize_ops )
Set default configurations for all ops in the model.
Set default configurations for all ops in the model .
"""
        Set default configurations for all ops in the model.
        :param default_op_configs: Default configurations for ops
        """ # Modified quantize ops is a dictionary that keeps track of the quantize ops that are modified by this function. # It is initialized as empty, and each time self._set_config_for_op is called, an op will be added. # If ever a quantize op's attribute has already been modified this round and a contradicting setting is # specified, an assertion will be thrown. # Don't set the input op setting, since it should be off by default even if all ops within the model # are to be quantized.  If model input is to be quantized, model_input should be set in config file # instead.
self
default_op_configs
ConfigType
Default configurations for ops
Default configurations for ops