From bf6c00c726eab6e101460dad935604c05c2596da Mon Sep 17 00:00:00 2001 From: Lester Solbakken Date: Wed, 13 May 2020 13:41:32 +0200 Subject: Remove floow in reshape --- .../java/ai/vespa/rankingexpression/importer/operations/Reshape.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'model-integration') 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))); } -- cgit v1.2.3