TrainingExtensions/tensorflow/src/python/aimet_tensorflow/quantsim_config/quantsim_config.py
def _get_quant_ops_for_input_setting(self, op: Op):
        """
        Get a list of quantize ops to modify when setting input quantization for the op.
        :param op: Op to set input quantization settings for.
        :return: List of quantize ops to modify
        """
        input_true_quant_ops = []
        queue = [op]
        processed_ops = set()
        while queue:
            curr_op = queue.pop()
            processed_ops.add(curr_op)
            for inp_op in curr_op.input_ops:
                if inp_op in self._op_to_quant_ops_dict:
                    _, output_quantize_op = self._op_to_quant_ops_dict[inp_op]
                    input_true_quant_ops.append(output_quantize_op)
                else:
                    if inp_op not in processed_ops:
                        queue.append(inp_op)
        return input_true_quant_ops

        Get a list of quantize ops to modify when setting input quantization for the op.
        :param op: Op to set input quantization settings for.
        :return: List of quantize ops to modify
        
Get a list of quantize ops to modify when setting input quantization for the op.
Get a list of quantize ops to modify when setting input quantization for the op .
def _get_quant_ops_for_input_setting(self, op: Op):
        input_true_quant_ops = []
        queue = [op]
        processed_ops = set()
        while queue:
            curr_op = queue.pop()
            processed_ops.add(curr_op)
            for inp_op in curr_op.input_ops:
                if inp_op in self._op_to_quant_ops_dict:
                    _, output_quantize_op = self._op_to_quant_ops_dict[inp_op]
                    input_true_quant_ops.append(output_quantize_op)
                else:
                    if inp_op not in processed_ops:
                        queue.append(inp_op)
        return input_true_quant_ops
def _get_quant_ops_for_input_setting ( self , op : Op ) : input_true_quant_ops = [ ] queue = [ op ] processed_ops = set ( ) while queue : curr_op = queue . pop ( ) processed_ops . add ( curr_op ) for inp_op in curr_op . input_ops : if inp_op in self . _op_to_quant_ops_dict : _ , output_quantize_op = self . _op_to_quant_ops_dict [ inp_op ] input_true_quant_ops . append ( output_quantize_op ) else : if inp_op not in processed_ops : queue . append ( inp_op ) return input_true_quant_ops
Get a list of quantize ops to modify when setting input quantization for the op.
Get a list of quantize ops to modify when setting input quantization for the op .
"""
        Get a list of quantize ops to modify when setting input quantization for the op.
        :param op: Op to set input quantization settings for.
        :return: List of quantize ops to modify
        """
List of quantize ops to modify
List of quantize ops to modify
self
op
Op
Op to set input quantization settings for.
Op to set input quantization settings for .