Quellcode durchsuchen

ONNX graph input and output descriptions

Lutz Roeder vor 8 Jahren
Ursprung
Commit
7800a36a55
1 geänderte Dateien mit 2 neuen und 0 gelöschten Zeilen
  1. 2 0
      src/onnx-model.js

+ 2 - 0
src/onnx-model.js

@@ -184,6 +184,7 @@ class OnnxGraph {
                         this._inputs.push({
                             id: valueInfo.name,
                             name: valueInfo.name,
+                            description: valueInfo.docString,
                             type: OnnxTensor.formatType(valueInfo.type)
                         });
                     }
@@ -201,6 +202,7 @@ class OnnxGraph {
                     return {
                         id: valueInfo.name,
                         name: valueInfo.name,
+                        description: valueInfo.docString,
                         type: OnnxTensor.formatType(valueInfo.type)
                     };
                 });