def METHOD_NAME(connected_graph: ConnectedGraph, param_name: str):
    """
    Gets op for a given param name
    :param connected_graph: Connected graph
    :param param_name: Name of the parameter
    """
    ops = connected_graph.get_all_ops()
    for op in ops.values():
        if op.parameters:
            for param, _ in op.parameters.values():
                if param.name == param_name:
                    return op
    return None