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 /config-model/src/test/java/com/yahoo/searchdefinition/processing/RankingExpressionWithOnnxTestCase.java | |
parent | 2041886aa197dff3a68abe5e821401f37adaa6f4 (diff) |
Add option to use default signature for onnx models
Diffstat (limited to 'config-model/src/test/java/com/yahoo/searchdefinition/processing/RankingExpressionWithOnnxTestCase.java')
-rw-r--r-- | config-model/src/test/java/com/yahoo/searchdefinition/processing/RankingExpressionWithOnnxTestCase.java | 16 |
1 files changed, 15 insertions, 1 deletions
diff --git a/config-model/src/test/java/com/yahoo/searchdefinition/processing/RankingExpressionWithOnnxTestCase.java b/config-model/src/test/java/com/yahoo/searchdefinition/processing/RankingExpressionWithOnnxTestCase.java index 754f161c70b..4387d19c474 100644 --- a/config-model/src/test/java/com/yahoo/searchdefinition/processing/RankingExpressionWithOnnxTestCase.java +++ b/config-model/src/test/java/com/yahoo/searchdefinition/processing/RankingExpressionWithOnnxTestCase.java @@ -128,6 +128,20 @@ public class RankingExpressionWithOnnxTestCase { } @Test + public void testOnnxReferenceWithSpecifiedOutput() { + RankProfileSearchFixture search = fixtureWith("tensor<float>(d0[2],d1[784])(0.0)", + "onnx('mnist_softmax.onnx', 'add')"); + search.assertFirstPhaseExpression(vespaExpression, "my_profile"); + } + + @Test + public void testOnnxReferenceWithSpecifiedOutputAndSignature() { + RankProfileSearchFixture search = fixtureWith("tensor<float>(d0[2],d1[784])(0.0)", + "onnx('mnist_softmax.onnx', 'default.add')"); + search.assertFirstPhaseExpression(vespaExpression, "my_profile"); + } + + @Test public void testOnnxReferenceMissingFunction() throws ParseException { try { RankProfileSearchFixture search = new RankProfileSearchFixture( @@ -171,7 +185,7 @@ public class RankingExpressionWithOnnxTestCase { @Test public void testOnnxReferenceSpecifyingNonExistingOutput() { try { - RankProfileSearchFixture search = fixtureWith("tensor(d0[2],d1[784])(0.0)", + RankProfileSearchFixture search = fixtureWith("tensor<float>(d0[2],d1[784])(0.0)", "onnx('mnist_softmax.onnx', 'y')"); search.assertFirstPhaseExpression(vespaExpression, "my_profile"); fail("Expecting exception"); |