Browse Source

Fix ONNX map graph output type rendering

Lutz Roeder 8 years ago
parent
commit
61d7de1946
2 changed files with 12 additions and 9 deletions
  1. 2 2
      src/view-onnx.js
  2. 10 7
      src/view-template.js

+ 2 - 2
src/view-onnx.js

@@ -82,7 +82,7 @@ class OnnxModel {
     }
 
     updateActiveGraph(name) {
-        this._activeGraph = (name == this._graph._graph.name) ? this._graph : null;
+        this._activeGraph = (name == this._graphs[0]._graph.name) ? this._graph : null;
     }
 }
 
@@ -565,7 +565,7 @@ class OnnxTensor {
         }
         else if (type.value == 'mapType') {
             var mapType = type.mapType;
-            return '<' + OnnxTensor.formatElementType(mapType.keyType) + ', ' + OnnxTensor.formatType(mapType.valueType) + '>';
+            return '<' + OnnxTensor.formatElementType(mapType.keyType) + ',' + OnnxTensor.formatType(mapType.valueType) + '>';
         }
         else if (!type.value) {
             return '?';

+ 10 - 7
src/view-template.js

@@ -124,7 +124,7 @@ var summaryTemplate = `
 <div class='border section'>
 <table>
 {{#properties}}
-<tr><td>{{{name}}}</td><td>{{{value}}}</td></tr>
+<tr><td>{{name}}</td><td>{{value}}</td></tr>
 {{/properties}}
 </table>
 </div>
@@ -133,20 +133,20 @@ var summaryTemplate = `
 <div class='border section'>
 <table>
 {{#if name}}
-<tr><td>Name</td><td>{{{name}}}</td></tr>
+<tr><td>Name</td><td>{{name}}</td></tr>
 {{/if}}
 {{#if version}}
-<tr><td>Version</td><td>{{{version}}}</td></tr>
+<tr><td>Version</td><td>{{version}}</td></tr>
 {{/if}}
 {{#if tags}}
-<tr><td>Tags</td><td>{{{tags}}}</td></tr>
+<tr><td>Tags</td><td>{{tags}}</td></tr>
 {{/if}}
 <tr>
 {{#if inputs}}
 <td>Inputs</td>
 <td>
 {{#inputs}}
-<code>{{{type}}}</code> : {{{name}}}<br>
+<code>{{type}}</code> : {{name}}<br>
 {{/inputs}}
 </td>
 </tr>
@@ -156,13 +156,16 @@ var summaryTemplate = `
 <td>Outputs</td>
 <td>
 {{#outputs}}
-<code>{{{type}}}</code> : {{{name}}}<br>
+<code>{{type}}</code> : {{name}}<br>
 {{/outputs}}
 </td>
 </tr>
 {{/if}}
 {{#if description}}
-<tr><td>{{{Description}}}</td><td>{{{description}}}</td></tr>
+<tr>
+<td>Description</td>
+<td>{{description}}</td>
+</tr>
 {{/if}}
 </table>
 <button style='float: right; width: 80px;' onclick='javascript:updateActiveGraph("{{{name}}}");'>View</button>