diff options
| author | Lester Solbakken <lesters@oath.com> | 2020-04-26 10:02:20 +0200 |
|---|---|---|
| committer | Lester Solbakken <lesters@oath.com> | 2020-04-26 10:02:20 +0200 |
| commit | 180cb44c805d3c2703041c0e34d2e4bfb8b67915 (patch) | |
| tree | 30937fe61e258fb2257dea750f44b943b42f103f /model-integration | |
| parent | 12b09c46a3e32df3a90d326f30171bc789d1bcb8 (diff) | |
Ensure dimensions to matmul over are innermost
Diffstat (limited to 'model-integration')
| -rw-r--r-- | model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/MatMul.java | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/MatMul.java b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/MatMul.java index 2b0af93fd8e..d326291a5a5 100644 --- a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/MatMul.java +++ b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/MatMul.java @@ -126,7 +126,7 @@ public class MatMul extends IntermediateOperation { // a1 < a2 < a3 < a4 for (int j = i+1; j < typeA.rank(); ++j) { String jDim = typeA.dimensionNames().get(j); - renamer.addConstraint(iDim, jDim, DimensionRenamer.Constraint.lessThan(true), this); + renamer.addConstraint(iDim, jDim, DimensionRenamer.Constraint.lessThan(false), this); } // not equal to last 2 dimensions in B for (int j = typeB.rank()-2; j < typeB.rank(); ++j) { @@ -148,7 +148,7 @@ public class MatMul extends IntermediateOperation { // b1 < b2 < b3 < b4 for (int j = i+1; j < typeB.rank(); ++j) { String jDim = typeB.dimensionNames().get(j); - renamer.addConstraint(iDim, jDim, DimensionRenamer.Constraint.lessThan(true), this); + renamer.addConstraint(iDim, jDim, DimensionRenamer.Constraint.lessThan(false), this); } // not equal to last 2 dimensions in A for (int j = typeA.rank()-2; j < typeA.rank(); ++j) { |