Просмотр исходного кода

ONNX graph input and output descriptions

Lutz Roeder 8 лет назад
Родитель
Сommit
7800a36a55
1 измененных файлов с 2 добавлено и 0 удалено
  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)
                     };
                 });