diff options
author | Lester Solbakken <lesters@users.noreply.github.com> | 2019-05-08 09:16:06 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2019-05-08 09:16:06 +0200 |
commit | 2c4c16f3d53b2aed1ddb25e456271ad1d97eb691 (patch) | |
tree | 56c94a87bee7a6804f0a2ed588abab35aba29126 /model-integration | |
parent | 0c379910129e489b8bbd3ecde16690120068a669 (diff) | |
parent | 4ef162353bc007ed3645de2b4008bc58110375ae (diff) |
Merge pull request #9307 from vespa-engine/bratseth/emit-float-tensors-in-config
Bratseth/emit float tensors in config
Diffstat (limited to 'model-integration')
-rw-r--r-- | model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxMnistSoftmaxImportTestCase.java | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxMnistSoftmaxImportTestCase.java b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxMnistSoftmaxImportTestCase.java index 07814687dc6..35c853bd746 100644 --- a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxMnistSoftmaxImportTestCase.java +++ b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxMnistSoftmaxImportTestCase.java @@ -31,13 +31,13 @@ public class OnnxMnistSoftmaxImportTestCase { Tensor constant0 = Tensor.from(model.largeConstants().get("test_Variable")); assertNotNull(constant0); - assertEquals(new TensorType.Builder().indexed("d2", 784).indexed("d1", 10).build(), + assertEquals(new TensorType.Builder(TensorType.Value.FLOAT).indexed("d2", 784).indexed("d1", 10).build(), constant0.type()); assertEquals(7840, constant0.size()); Tensor constant1 = Tensor.from(model.largeConstants().get("test_Variable_1")); assertNotNull(constant1); - assertEquals(new TensorType.Builder().indexed("d1", 10).build(), constant1.type()); + assertEquals(new TensorType.Builder(TensorType.Value.FLOAT).indexed("d1", 10).build(), constant1.type()); assertEquals(10, constant1.size()); // Check inputs @@ -52,7 +52,7 @@ public class OnnxMnistSoftmaxImportTestCase { output.expression()); assertEquals(TensorType.fromSpec("tensor<float>(d0[],d1[784])"), model.inputs().get(model.defaultSignature().inputs().get("Placeholder"))); - assertEquals("{Placeholder=tensor(d0[],d1[784])}", output.argumentTypes().toString()); + assertEquals("{Placeholder=tensor<float>(d0[],d1[784])}", output.argumentTypes().toString()); } @Test |