TrainingExtensions/torch/src/python/aimet_torch/quantsim_config/quantsim_config.py
def _get_tensor_quantizers_for_output_false_setting(self, op: Op) -> List[TensorQuantizer]:
        """
        Get a list of tensor quantizers that would be affected if the given op is specified to have output quantization
        disabled.
        :param op: Op to disable output quantization for
        :return: List of tensor quantizers that would be affected if the given op is specified to have output
        quantization disabled.
        """
        tensor_quantizers_for_output_false = []
        neighboring_output_ops = _get_all_ops_in_neighborhood(op, 'output')
        for output_op in neighboring_output_ops:
            if output_op.type != 'Split' and output_op.get_module() is not None and output_op.get_module() in \
                    self._module_to_quantsim_wrapper_dict:
                qc_quantize_wrapper = self._module_to_quantsim_wrapper_dict[output_op.get_module()]
                if neighboring_output_ops[output_op] == 'input':
                    tensor_quantizers_for_output_false.append(qc_quantize_wrapper.input_quantizer)
                else:
                    tensor_quantizers_for_output_false.append(qc_quantize_wrapper.output_quantizers[0])
        return tensor_quantizers_for_output_false

        Get a list of tensor quantizers that would be affected if the given op is specified to have output quantization
        disabled.
        :param op: Op to disable output quantization for
        :return: List of tensor quantizers that would be affected if the given op is specified to have output
        quantization disabled.
        
Get a list of tensor quantizers that would be affected if the given op is specified to have output quantization
disabled.
Get a list of tensor quantizers that would be affected if the given op is specified to have output quantization disabled .
def _get_tensor_quantizers_for_output_false_setting(self, op: Op) -> List[TensorQuantizer]:
        tensor_quantizers_for_output_false = []
        neighboring_output_ops = _get_all_ops_in_neighborhood(op, 'output')
        for output_op in neighboring_output_ops:
            if output_op.type != 'Split' and output_op.get_module() is not None and output_op.get_module() in \
                    self._module_to_quantsim_wrapper_dict:
                qc_quantize_wrapper = self._module_to_quantsim_wrapper_dict[output_op.get_module()]
                if neighboring_output_ops[output_op] == 'input':
                    tensor_quantizers_for_output_false.append(qc_quantize_wrapper.input_quantizer)
                else:
                    tensor_quantizers_for_output_false.append(qc_quantize_wrapper.output_quantizers[0])
        return tensor_quantizers_for_output_false
def _get_tensor_quantizers_for_output_false_setting ( self , op : Op ) -> List [ TensorQuantizer ] : tensor_quantizers_for_output_false = [ ] neighboring_output_ops = _get_all_ops_in_neighborhood ( op , 'output' ) for output_op in neighboring_output_ops : if output_op . type != 'Split' and output_op . get_module ( ) is not None and output_op . get_module ( ) in self . _module_to_quantsim_wrapper_dict : qc_quantize_wrapper = self . _module_to_quantsim_wrapper_dict [ output_op . get_module ( ) ] if neighboring_output_ops [ output_op ] == 'input' : tensor_quantizers_for_output_false . append ( qc_quantize_wrapper . input_quantizer ) else : tensor_quantizers_for_output_false . append ( qc_quantize_wrapper . output_quantizers [ 0 ] ) return tensor_quantizers_for_output_false
Get a list of tensor quantizers that would be affected if the given op is specified to have output quantization
disabled.
Get a list of tensor quantizers that would be affected if the given op is specified to have output quantization disabled .
"""
        Get a list of tensor quantizers that would be affected if the given op is specified to have output quantization
        disabled.
        :param op: Op to disable output quantization for
        :return: List of tensor quantizers that would be affected if the given op is specified to have output
        quantization disabled.
        """
List of tensor quantizers that would be affected if the given op is specified to have output
quantization disabled.
List of tensor quantizers that would be affected if the given op is specified to have output quantization disabled .
self
op
Op
Op to disable output quantization for
Op to disable output quantization for