in toolings/tfjs-debugger/src/app/data_model/run_results.ts [103:121]
export function modelJsonToModelGraph(json: ModelJson): ModelGraph {
const modelGraph: ModelGraph = {};
for (const node of json.modelTopology.node) {
if (!node.name) {
continue;
}
modelGraph[node.name] = {
// Use node name as id since it is unique.
id: node.name,
op: node.op || '',
// TODO: calculate node size based on node's op and other factors.
width: 60,
height: 30,
inputNodeIds: node.input || [],
};
}
return modelGraph;
}