search

microsoft / onnxscript

ONNX Script enables developers to naturally author ONNX functions and models using a subset of Python.
https://onnxscript.ai/
MIT License
285 stars 54 forks source link

Graph tags #347

Open justinchuby opened 1 year ago

justinchuby commented 1 year ago

Tag nodes with labels so we can group and track them.

justinchuby commented 1 year ago
a -> [b - > {c, d}] (graph1.onnx) a -> [e] (graph2.onnx)

a -> [f -> g] (graph3.onnx)

{b, c, d, e}: transformation_pass/fusion/group/1 {e, f, g}: transformation_pass/expand/group/1 {a} -> role/input

transformation_pass/fusion/group/1 -> "energy efficiency improvement"

{b, c, d}: module/class/Attention

justinchuby commented 1 year ago

We may need to use a fixd size identifier for tags. Maybe a hash of some sort?