diff options
author | Jon Bratseth <bratseth@verizonmedia.com> | 2019-11-05 22:49:08 +0100 |
---|---|---|
committer | Jon Bratseth <bratseth@verizonmedia.com> | 2019-11-05 22:49:08 +0100 |
commit | ed8c274dc76794efa692efba6cf509b058b13648 (patch) | |
tree | c1dcb9fbc70b851be5cfdb8c335089283715f698 /config-model | |
parent | 64c5daa351557869e64786188afa75ed3b59991b (diff) |
Literal tensors with value expressions
Diffstat (limited to 'config-model')
-rw-r--r-- | config-model/src/main/java/com/yahoo/vespa/model/ml/ConvertedModel.java | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/config-model/src/main/java/com/yahoo/vespa/model/ml/ConvertedModel.java b/config-model/src/main/java/com/yahoo/vespa/model/ml/ConvertedModel.java index 4acb47df179..bbfd2004caa 100644 --- a/config-model/src/main/java/com/yahoo/vespa/model/ml/ConvertedModel.java +++ b/config-model/src/main/java/com/yahoo/vespa/model/ml/ConvertedModel.java @@ -432,7 +432,7 @@ public class ConvertedModel { if (node instanceof ReferenceNode) { ReferenceNode referenceNode = (ReferenceNode) node; if (model.inputTypeSpec(referenceNode.getName()).isPresent()) { - return reduceBatchDimensionExpression(TensorFunctionNode.wrapArgument(node), typeContext); + return reduceBatchDimensionExpression(TensorFunctionNode.wrap(node), typeContext); } } if (node instanceof CompositeNode) { @@ -485,7 +485,7 @@ public class ConvertedModel { new GeneratorLambdaFunctionNode(expandDimensionsType, generatedExpression) .asLongListToDoubleOperator()); - Join expand = new Join(TensorFunctionNode.wrapArgument(node), generatedFunction, ScalarFunctions.multiply()); + Join expand = new Join(TensorFunctionNode.wrap(node), generatedFunction, ScalarFunctions.multiply()); return new TensorFunctionNode(expand); } return node; |