From 07c7d7c29a960940213dba2db6f24c63353a0064 Mon Sep 17 00:00:00 2001 From: Arne Juul Date: Fri, 17 Mar 2023 10:34:21 +0000 Subject: handle DynamicTensor with embedded expressions --- .../java/com/yahoo/schema/expressiontransforms/InputRecorder.java | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'config-model/src') diff --git a/config-model/src/main/java/com/yahoo/schema/expressiontransforms/InputRecorder.java b/config-model/src/main/java/com/yahoo/schema/expressiontransforms/InputRecorder.java index 993a2442fdb..84d0997b12e 100644 --- a/config-model/src/main/java/com/yahoo/schema/expressiontransforms/InputRecorder.java +++ b/config-model/src/main/java/com/yahoo/schema/expressiontransforms/InputRecorder.java @@ -12,6 +12,7 @@ import com.yahoo.searchlib.rankingexpression.rule.ExpressionNode; import com.yahoo.searchlib.rankingexpression.rule.ReferenceNode; import com.yahoo.searchlib.rankingexpression.rule.TensorFunctionNode; import com.yahoo.searchlib.rankingexpression.transform.ExpressionTransformer; +import com.yahoo.tensor.functions.DynamicTensor; import com.yahoo.tensor.functions.Generate; import com.yahoo.tensor.functions.Slice; @@ -54,6 +55,13 @@ public class InputRecorder extends ExpressionTransformer { } return transformChildren(t, childContext); } + if (f instanceof DynamicTensor d) { + for (var tf : d.cellGeneratorFunctions()) { + if (tf instanceof TensorFunctionNode.ExpressionTensorFunction expr) { + transform(expr.wrappedExpression(), context); + } + } + } if (f instanceof Slice s) { for (var tf : s.selectorFunctions()) { if (tf instanceof TensorFunctionNode.ExpressionTensorFunction expr) { -- cgit v1.2.3