handle DynamicTensor

1 files changed, 8 insertions, 0 deletions

diff --git a/model-evaluation/src/main/java/ai/vespa/models/evaluation/BindingExtractor.java b/model-evaluation/src/main/java/ai/vespa/models/evaluation/BindingExtractor.java
index 126e9f9f4e6..bc71e51655c 100644
--- a/model-evaluation/src/main/java/ai/vespa/models/evaluation/BindingExtractor.java
+++ b/model-evaluation/src/main/java/ai/vespa/models/evaluation/BindingExtractor.java
@@ -8,6 +8,7 @@ import com.yahoo.searchlib.rankingexpression.rule.ConstantNode;
 import com.yahoo.searchlib.rankingexpression.rule.ExpressionNode;
 import com.yahoo.searchlib.rankingexpression.rule.ReferenceNode;
 import com.yahoo.searchlib.rankingexpression.rule.TensorFunctionNode;
+import com.yahoo.tensor.functions.DynamicTensor;
 import com.yahoo.tensor.functions.Generate;
 import com.yahoo.tensor.functions.Slice;
 
@@ -109,6 +110,13 @@ class BindingExtractor {
                     }
                 }
             }
+            else if (f instanceof DynamicTensor<?> d) {
+                for (var tf : d.cellGeneratorFunctions()) {
+                    if (tf instanceof TensorFunctionNode.ExpressionTensorFunction expr) {
+                        result.merge(extractBindTargets(expr.wrappedExpression()));
+                    }
+                }
+            }
             return result;
         }
         else if (isOnnx(node)) {