Function _annotate_nodes

backends/example/example_operators/utils.py:23–34

(node_tuples, quant_spec, input_node=False)

def _annotate_nodes(node_tuples, quant_spec, input_node=False):
    for node_tuple in node_tuples:
        node = node_tuple[0]
        quant_annotation = node.meta.get(
            Q_ANNOTATION_KEY, QuantizationAnnotation(_annotated=True)
        )
        if input_node:
            input_node = node_tuple[1]
            quant_annotation.input_qspec_map[input_node] = quant_spec
        else:
            quant_annotation.output_qspec = quant_spec
        node.meta[Q_ANNOTATION_KEY] = quant_annotation