diff options
author | Lester Solbakken <lesters@oath.com> | 2020-08-31 09:08:26 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2020-09-04 08:30:25 +0200 |
commit | 67caf3b6eee690bcd0c7fc7a7666bd2cf41b8816 (patch) | |
tree | f0b424c80c357091d738831bfcce16d071f006a1 /model-integration/src/main/java/ai/vespa/rankingexpression/importer/ModelImporter.java | |
parent | 9a735030cd58e7f7ce7c2cd9bcaae121089e6ee7 (diff) |
Import TensorFlow models via ONNX conversion
Diffstat (limited to 'model-integration/src/main/java/ai/vespa/rankingexpression/importer/ModelImporter.java')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/rankingexpression/importer/ModelImporter.java | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/ModelImporter.java b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/ModelImporter.java index a9d71b7d9d5..8f73cd02184 100644 --- a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/ModelImporter.java +++ b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/ModelImporter.java @@ -74,7 +74,7 @@ public abstract class ModelImporter implements MlModelImporter { signature.input(input.getKey(), input.getValue()); } for (Map.Entry<String, String> output : graph.outputs(signatureName).entrySet()) { - signature.output(output.getKey(), output.getValue()); + signature.output(IntermediateOperation.vespaName(output.getKey()), output.getValue()); } } } |