Explorar el Código

ONNX graph input and output descriptions

Lutz Roeder hace 8 años
padre
commit
7800a36a55
Se han modificado 1 ficheros con 2 adiciones y 0 borrados
  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)
                     };
                 });