Pull request pytorch#34: Add shapes and types to links in ExportedProgram visualization

Merge in AITEC/executorch from feature/nxf93343/program-vizualizer-add-shapes-and-types to main-nxp

* commit '1833e0349b9a62e382b42911f8a6ced167eaa984':
  Add shapes and types to link in ExportedProgram visualization

Author: skywall

backends/nxp/tests/exported_program_vizualize.py

@@ -61,7 +61,16 @@ def name_color(string):  # pseudo-randomization function
     # Add connections between nodes
     for node in exported_program.graph.nodes:
         for user in node.users:
-            graph.newLink(graph_items[node.name], graph_items[user.name])
+            link = graph.newLink(graph_items[node.name], graph_items[user.name])
+
+            label = ""
+            if "val" in node.meta:
+                tensor = node.meta["val"]
+                if isinstance(tensor, tuple):
+                    tensor = tensor[0]  # Fake tensor
+                label = f"  ({list(tensor.shape)} | {tensor.dtype})"
+
+            graph.propertyAppend(link, "label", label)
 
     with open("graph.dot", "w") as f:
         graph.dot(f)