diff options
author | Lester Solbakken <lesters@oath.com> | 2019-11-22 13:01:49 +0100 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2019-11-22 13:01:49 +0100 |
commit | 9546f2eca2d571947b7c473c5e69c71a1c9d2505 (patch) | |
tree | fac7672712f34a6401afc6de750ac9d2d99f5463 /model-integration | |
parent | 1d25742477886696da56cd99faf7766442f67038 (diff) |
Add correct name for input to onnx intermediate graph
Diffstat (limited to 'model-integration')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/rankingexpression/importer/onnx/GraphImporter.java | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/onnx/GraphImporter.java b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/onnx/GraphImporter.java index 4fa6c09c636..280fe354149 100644 --- a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/onnx/GraphImporter.java +++ b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/onnx/GraphImporter.java @@ -125,7 +125,7 @@ class GraphImporter { operation = new Argument(intermediateGraph.name(), valueInfoProto.getName(), type); intermediateGraph.inputs(intermediateGraph.defaultSignature()) - .put(IntermediateOperation.namePartOf(name), operation.name()); + .put(IntermediateOperation.namePartOf(name), operation.vespaName()); } else if (isConstantTensor(name, onnxGraph)) { Onnx.TensorProto tensorProto = getConstantTensor(name, onnxGraph); |