diff options
author | Lester Solbakken <lesters@oath.com> | 2020-05-13 13:41:32 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2020-05-13 13:41:32 +0200 |
commit | bf6c00c726eab6e101460dad935604c05c2596da (patch) | |
tree | 8a44271c84bdec60089b44a8c1e2c2909afb97a6 /model-integration/src/main | |
parent | 798b6768616d74a274677cd5be3671a3acea9244 (diff) |
Remove floow in reshape
Diffstat (limited to 'model-integration/src/main')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Reshape.java | 2 |
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))); } |