fix: include graph outputs

Author: jmahbs

src/vgf_adapter_model_explorer/spirv/spirv.py

@@ -70,7 +70,7 @@ def build_function_graph(
             traversed_ops=traversed_ops,
         )
 
-        if "Tosa" not in operation.name:
+        if "Tosa" not in operation.name and "GraphOutputs" not in operation.name:
             continue
 
         function_graph.nodes.append(node)

src/vgf_adapter_model_explorer/tests/fixtures/mobilenet_v2/expected.json

@@ -32129,6 +32129,39 @@
           "style": null,
           "config": null
         },
+        {
+          "id": "1583",
+          "label": "GraphOutputs",
+          "namespace": "graph_partition_0",
+          "subgraphIds": [],
+          "attrs": [
+            {
+              "key": "loc",
+              "value": "loc(\"-\":1589:5)"
+            }
+          ],
+          "incomingEdges": [
+            {
+              "sourceNodeId": "1582",
+              "sourceNodeOutputId": "0",
+              "targetNodeInputId": "0"
+            }
+          ],
+          "outputsMetadata": [],
+          "inputsMetadata": [
+            {
+              "id": "0",
+              "attrs": [
+                {
+                  "key": "tensor_shape",
+                  "value": "!spirv.arm.tensor<1x1001xi8>"
+                }
+              ]
+            }
+          ],
+          "style": null,
+          "config": null
+        },
         {
           "id": "mrt_115",
           "label": "VK_DESCRIPTOR_TYPE_TENSOR_ARM",
@@ -32178,7 +32211,7 @@
           "attrs": [],
           "incomingEdges": [
             {
-              "sourceNodeId": "1582",
+              "sourceNodeId": "1583",
               "sourceNodeOutputId": "0",
               "targetNodeInputId": "0"
             }

src/vgf_adapter_model_explorer/tests/fixtures/vww4/expected.json

@@ -29998,6 +29998,39 @@
           "style": null,
           "config": null
         },
+        {
+          "id": "1523",
+          "label": "GraphOutputs",
+          "namespace": "graph_partition_0",
+          "subgraphIds": [],
+          "attrs": [
+            {
+              "key": "loc",
+              "value": "loc(\"-\":1529:5)"
+            }
+          ],
+          "incomingEdges": [
+            {
+              "sourceNodeId": "1522",
+              "sourceNodeOutputId": "0",
+              "targetNodeInputId": "0"
+            }
+          ],
+          "outputsMetadata": [],
+          "inputsMetadata": [
+            {
+              "id": "0",
+              "attrs": [
+                {
+                  "key": "tensor_shape",
+                  "value": "!spirv.arm.tensor<1x2xi8>"
+                }
+              ]
+            }
+          ],
+          "style": null,
+          "config": null
+        },
         {
           "id": "mrt_95",
           "label": "VK_DESCRIPTOR_TYPE_TENSOR_ARM",
@@ -30047,7 +30080,7 @@
           "attrs": [],
           "incomingEdges": [
             {
-              "sourceNodeId": "1522",
+              "sourceNodeId": "1523",
               "sourceNodeOutputId": "0",
               "targetNodeInputId": "0"
             }