diff options
author | Arne Juul <arnej@yahooinc.com> | 2023-03-20 12:19:34 +0000 |
---|---|---|
committer | Arne Juul <arnej@yahooinc.com> | 2023-03-20 12:22:52 +0000 |
commit | baa9c48be8732564e00730efe680df26a8f47f4c (patch) | |
tree | 9c83c10f0a225cdd5828013b56ffee6556d76a62 /model-evaluation | |
parent | 5791e60dfcd5f83d0e77e45498318eeb3dd33ee3 (diff) |
use withTransformedExpressions for wiring
Diffstat (limited to 'model-evaluation')
-rw-r--r-- | model-evaluation/src/main/java/ai/vespa/models/evaluation/BindingExtractor.java | 19 |
1 files changed, 5 insertions, 14 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 bc71e51655c..8cdd5387acd 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 @@ -96,6 +96,11 @@ class BindingExtractor { for (ExpressionNode child : tfn.children()) { result.merge(extractBindTargets(child)); } + // ignore return value: + tfn.withTransformedExpressions(expr -> { + result.merge(extractBindTargets(expr)); + return expr; + }); var f = tfn.function(); if (f instanceof Generate) { var tt = f.type(null); @@ -103,20 +108,6 @@ class BindingExtractor { result.removeTarget(dim.name()); } } - else if (f instanceof Slice<?> s) { - for (var selectorFunc : s.selectorFunctions()) { - if (selectorFunc instanceof TensorFunctionNode.ExpressionTensorFunction expr) { - result.merge(extractBindTargets(expr.wrappedExpression())); - } - } - } - 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)) { |