diff options
Diffstat (limited to 'model-evaluation')
| -rw-r--r-- | model-evaluation/src/main/java/ai/vespa/models/evaluation/BindingExtractor.java | 8 |
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)) { |