torch.fx.experimental.symbolic_shapes.find_symbol_binding_fx_nodes

torch.fx.experimental.symbolic_shapes.find_symbol_binding_fx_nodes(graph)

Find all nodes in an FX graph that bind sympy Symbols.

This function scans through all nodes in the given FX graph and identifies nodes that bind sympy Symbols (typically placeholder nodes with SymInt values). When multiple nodes bind the same symbol, only the first occurrence is kept.

Parameters:

graph (Graph) – The FX graph to search for symbol binding nodes

Returns:

A dictionary mapping from sympy Symbols to their binding FX nodes

Return type:

dict[Symbol, Node]