diff options
author | Lester Solbakken <lesters@oath.com> | 2020-06-24 14:51:20 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2020-06-24 14:51:20 +0200 |
commit | bb58df8c94621ee5274e4ee00d2daab507662c21 (patch) | |
tree | 7be84a0f30f2262a13309a68ccaa8d0e56d10e67 /model-integration | |
parent | 6821733667528dddeed658205358f6b9dda12090 (diff) |
Use correct value type for onnx range operator
Diffstat (limited to 'model-integration')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Range.java | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Range.java b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Range.java index 6df686cf910..18018a1a73a 100644 --- a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Range.java +++ b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Range.java @@ -51,7 +51,7 @@ public class Range extends IntermediateOperation { delta = getConstantInput(2, "delta"); elements = (long) Math.ceil((limit - start) / delta); - OrderedTensorType type = new OrderedTensorType.Builder() + OrderedTensorType type = new OrderedTensorType.Builder(inputs.get(0).type().get().type().valueType()) .add(TensorType.Dimension.indexed(vespaName(), elements)) .build(); return type; |