diff options
Diffstat (limited to 'model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Shape.java')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Shape.java | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Shape.java b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Shape.java index 79f3012c327..57175092b5c 100644 --- a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Shape.java +++ b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Shape.java @@ -37,6 +37,11 @@ public class Shape extends IntermediateOperation { return true; } + @Override + public Shape withInputs(List<IntermediateOperation> inputs) { + return new Shape(modelName(), name(), inputs); + } + private void createConstantValue() { if (!allInputTypesPresent(1)) { return; @@ -50,4 +55,7 @@ public class Shape extends IntermediateOperation { this.setConstantValue(new TensorValue(builder.build())); } + @Override + public String operationName() { return "Shape"; } + } |