Function find_node_by_target

backends/mlx/test/test_pattern_utils.py:24–29

Find first call_function node whose target contains target_name.

(graph, target_name)

def find_node_by_target(graph, target_name):
    """Find first call_function node whose target contains target_name."""
    for node in graph.nodes:
        if node.op == "call_function" and target_name in str(node.target):
            return node
    return None


def find_all_nodes_by_target(graph, target_name):