diff options
author | Henning Baldersheim <balder@yahoo-inc.com> | 2022-11-09 20:16:11 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-11-09 20:16:11 +0100 |
commit | ba3ac72d3146d9e64a4bf32c993fd4969b272c7d (patch) | |
tree | b9e57f781bde4d962835b4ffdc56a58e5d7585cd /model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxMnistSoftmaxImportTestCase.java | |
parent | 606bc6b517e09bba9f36a4ef6506a588d1254bd4 (diff) | |
parent | ec1413c6baca2e30bdd732ca8e20a108c9e66852 (diff) |
Merge pull request #24814 from vespa-engine/revert-24813-revert-24773-revert-24760-balder/model-importing-code-in-config-model-3v8.82.24
Revert "Revert "Revert "Balder/model importing code in config model [run-systemtest]"""
Diffstat (limited to 'model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/OnnxMnistSoftmaxImportTestCase.java')
-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 7fb167ee6f1..b6b63912c52 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 @@ -21,15 +21,15 @@ public class OnnxMnistSoftmaxImportTestCase { ImportedModel model = new OnnxImporter().importModel("test", "src/test/models/onnx/mnist_softmax/mnist_softmax.onnx").asNative(); // Check constants - assertEquals(2, model.largeConstantTensors().size()); + assertEquals(2, model.largeConstants().size()); - Tensor constant0 = model.largeConstantTensors().get("test_Variable"); + Tensor constant0 = Tensor.from(model.largeConstants().get("test_Variable")); assertNotNull(constant0); assertEquals(new TensorType.Builder(TensorType.Value.FLOAT).indexed("d2", 784).indexed("d1", 10).build(), constant0.type()); assertEquals(7840, constant0.size()); - Tensor constant1 = model.largeConstantTensors().get("test_Variable_1"); + Tensor constant1 = Tensor.from(model.largeConstants().get("test_Variable_1")); assertNotNull(constant1); assertEquals(new TensorType.Builder(TensorType.Value.FLOAT).indexed("d1", 10).build(), constant1.type()); assertEquals(10, constant1.size()); |