def get_qdq_to_remove(model, relu_input):
    """
    Return two lists, one for 'dq' and one for 'q'.
    """
    q_nodes_to_remove = []
    dq_nodes_to_remove = []
    q_nodes_output_to_remove = []
    for node in model.graph.node:
        if node.op_type == "QuantizeLinear" and node.input[0] in relu_input.values():
            q_nodes_to_remove.append(node)
            q_nodes_output_to_remove.append(node.output[0])
    for node in model.graph.node:
        if node.op_type == "DequantizeLinear" and node.input[0] in q_nodes_output_to_remove:
            dq_nodes_to_remove.append(node)
    return dq_nodes_to_remove, q_nodes_to_remove

    Return two lists, one for 'dq' and one for 'q'.
    
olive/passes/onnx/vitis_ai/quant_utils.py