google-ai-edge / model-explorer

A modern model graph visualizer and debugger
Apache License 2.0
960 stars 68 forks source link

Would be great to match connected nodes with regex search #72

Open maderix opened 2 months ago

maderix commented 2 months ago

While the node search with regex works great, It may be quite useful for optimization purposes to locate certain patterns in the graph. for example : Transpose->Reshapes in the model patterns which could be detected and optimized by inference libraries. This sort of search mechanism could be useful for a developer who's optimizing the graph.

Idea is to add neighborhood connectivity regex search where a regex like Matmul->.->Softmax* could be located and highlighted.

pkgoogle commented 2 months ago

I like this idea, might be easier to handle the single input/output case for now. We'll see where we can fit this in the roadmap.