diff options
author | Lester Solbakken <lesters@oath.com> | 2019-12-18 10:47:55 +0100 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2019-12-18 10:47:55 +0100 |
commit | 6af40601ac82204b3732096dee2ba23fe237e6e8 (patch) | |
tree | 69bca5b6a9aca95312859924ba49bcf7571f7975 /model-integration/src/test | |
parent | 2041886aa197dff3a68abe5e821401f37adaa6f4 (diff) |
Add option to use default signature for onnx models
Diffstat (limited to 'model-integration/src/test')
-rw-r--r-- | model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxOperationsTestCase.java | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxOperationsTestCase.java b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxOperationsTestCase.java index 6954abe5157..9927f4df04f 100644 --- a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxOperationsTestCase.java +++ b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxOperationsTestCase.java @@ -391,7 +391,7 @@ public class OnnxOperationsTestCase { private TensorFunction optimizeAndRename(String opName, IntermediateOperation op) { IntermediateGraph graph = new IntermediateGraph(modelName); graph.put(opName, op); - graph.outputs(graph.defaultSignature()).put(opName, opName); + graph.outputs(IntermediateGraph.defaultSignature()).put(opName, opName); graph.optimize(); return op.function().get(); } |