Skip to content

Commit 2fa3c80

Browse files
MartinPavellaMartinPavella
committed
NXP backend: Update the ExecuTorch graph visualizer to suite NXP use-cases.
Add style .json file, which can be loaded in the Model Explorer GUI, and it changes the visualization to highlight the clusters and partitions in different colors.
1 parent 2c706f1 commit 2fa3c80

File tree

2 files changed

+377
-5
lines changed

2 files changed

+377
-5
lines changed

‎devtools/visualization/model_explorer_styles/cluster_highlight_style.json‎

Lines changed: 236 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,236 @@
1+
[
2+
{
3+
"queries": [
4+
{
5+
"type": "node_type",
6+
"nodeType": "op_nodes"
7+
},
8+
{
9+
"type": "regex",
10+
"queryRegex": "quantize",
11+
"matchTypes": [
12+
"title"
13+
]
14+
}
15+
],
16+
"nodeType": "op_nodes",
17+
"styles": {
18+
"node_bg_color": {
19+
"id": "node_bg_color",
20+
"value": "#dce9e9"
21+
}
22+
},
23+
"version": "v2"
24+
},
25+
{
26+
"queries": [
27+
{
28+
"type": "node_type",
29+
"nodeType": "op_nodes"
30+
},
31+
{
32+
"type": "regex",
33+
"queryRegex": "aten.",
34+
"matchTypes": [
35+
"title"
36+
]
37+
}
38+
],
39+
"nodeType": "op_nodes",
40+
"styles": {
41+
"node_bg_color": {
42+
"id": "node_bg_color",
43+
"value": "#b4e3f5"
44+
}
45+
},
46+
"version": "v2"
47+
},
48+
{
49+
"queries": [
50+
{
51+
"type": "node_type",
52+
"nodeType": "layer_nodes"
53+
},
54+
{
55+
"type": "regex",
56+
"queryRegex": "cluster",
57+
"matchTypes": [
58+
"title"
59+
]
60+
}
61+
],
62+
"nodeType": "op_nodes",
63+
"styles": {
64+
"node_bg_color": {
65+
"id": "node_bg_color",
66+
"value": "#d0eae9"
67+
},
68+
"node_border_color": {
69+
"id": "node_border_color",
70+
"value": "#ffffff"
71+
}
72+
},
73+
"version": "v2"
74+
},
75+
{
76+
"queries": [
77+
{
78+
"type": "node_type",
79+
"nodeType": "layer_nodes"
80+
},
81+
{
82+
"type": "regex",
83+
"queryRegex": "partition 0",
84+
"matchTypes": [
85+
"title"
86+
]
87+
}
88+
],
89+
"nodeType": "op_nodes",
90+
"styles": {
91+
"node_bg_color": {
92+
"id": "node_bg_color",
93+
"value": "#fff1d5"
94+
}
95+
},
96+
"version": "v2"
97+
},
98+
{
99+
"queries": [
100+
{
101+
"type": "node_type",
102+
"nodeType": "layer_nodes"
103+
},
104+
{
105+
"type": "regex",
106+
"queryRegex": "partition 1",
107+
"matchTypes": [
108+
"title"
109+
]
110+
}
111+
],
112+
"nodeType": "op_nodes",
113+
"styles": {
114+
"node_bg_color": {
115+
"id": "node_bg_color",
116+
"value": "#fdffcc"
117+
}
118+
},
119+
"version": "v2"
120+
},
121+
{
122+
"queries": [
123+
{
124+
"type": "node_type",
125+
"nodeType": "layer_nodes"
126+
},
127+
{
128+
"type": "regex",
129+
"queryRegex": "partition 2",
130+
"matchTypes": [
131+
"title"
132+
]
133+
}
134+
],
135+
"nodeType": "op_nodes",
136+
"styles": {
137+
"node_bg_color": {
138+
"id": "node_bg_color",
139+
"value": "#ccffcc"
140+
}
141+
},
142+
"version": "v2"
143+
},
144+
{
145+
"queries": [
146+
{
147+
"type": "node_type",
148+
"nodeType": "layer_nodes"
149+
},
150+
{
151+
"type": "regex",
152+
"queryRegex": "partition 3",
153+
"matchTypes": [
154+
"title"
155+
]
156+
}
157+
],
158+
"nodeType": "op_nodes",
159+
"styles": {
160+
"node_bg_color": {
161+
"id": "node_bg_color",
162+
"value": "#ccffff"
163+
}
164+
},
165+
"version": "v2"
166+
},
167+
{
168+
"queries": [
169+
{
170+
"type": "node_type",
171+
"nodeType": "layer_nodes"
172+
},
173+
{
174+
"type": "regex",
175+
"queryRegex": "partition 4",
176+
"matchTypes": [
177+
"title"
178+
]
179+
}
180+
],
181+
"nodeType": "op_nodes",
182+
"styles": {
183+
"node_bg_color": {
184+
"id": "node_bg_color",
185+
"value": "#ffc6e2"
186+
}
187+
},
188+
"version": "v2"
189+
},
190+
{
191+
"queries": [
192+
{
193+
"type": "node_type",
194+
"nodeType": "layer_nodes"
195+
},
196+
{
197+
"type": "regex",
198+
"queryRegex": "partition 5",
199+
"matchTypes": [
200+
"title"
201+
]
202+
}
203+
],
204+
"nodeType": "op_nodes",
205+
"styles": {
206+
"node_bg_color": {
207+
"id": "node_bg_color",
208+
"value": "#ffcaff"
209+
}
210+
},
211+
"version": "v2"
212+
},
213+
{
214+
"queries": [
215+
{
216+
"type": "node_type",
217+
"nodeType": "layer_nodes"
218+
},
219+
{
220+
"type": "regex",
221+
"queryRegex": "partition 6",
222+
"matchTypes": [
223+
"title"
224+
]
225+
}
226+
],
227+
"nodeType": "op_nodes",
228+
"styles": {
229+
"node_bg_color": {
230+
"id": "node_bg_color",
231+
"value": "#d7d7ff"
232+
}
233+
},
234+
"version": "v2"
235+
}
236+
]

0 commit comments

Comments
 (0)