summaryrefslogtreecommitdiffstats
path: root/model-integration
diff options
context:
space:
mode:
authorLester Solbakken <lesters@oath.com>2020-05-13 13:41:32 +0200
committerLester Solbakken <lesters@oath.com>2020-05-13 13:41:32 +0200
commitbf6c00c726eab6e101460dad935604c05c2596da (patch)
tree8a44271c84bdec60089b44a8c1e2c2909afb97a6 /model-integration
parent798b6768616d74a274677cd5be3671a3acea9244 (diff)
Remove floow in reshape
Diffstat (limited to 'model-integration')
-rw-r--r--model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Reshape.java2
1 files changed, 1 insertions, 1 deletions
diff --git a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Reshape.java b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Reshape.java
index d8e806ae7e4..78eb5a0e7bb 100644
--- a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Reshape.java
+++ b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Reshape.java
@@ -166,7 +166,7 @@ public class Reshape extends IntermediateOperation {
ExpressionNode previousSize = new ConstantNode(new DoubleValue(previousInnerSize));
ExpressionNode mod = new ArithmeticNode(unrolled, ArithmeticOperator.MODULO, previousSize);
ExpressionNode div = new ArithmeticNode(new EmbracedNode(mod), ArithmeticOperator.DIVIDE, size);
- inputDimensionExpression = new EmbracedNode(new FunctionNode(Function.floor, div));
+ inputDimensionExpression = new EmbracedNode(div);
}
dimensionValues.add(new com.yahoo.tensor.functions.Slice.DimensionValue<>(Optional.of(inputDimensionName), wrapScalar(inputDimensionExpression)));
}