aboutsummaryrefslogtreecommitdiffstats
path: root/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/rule/TensorFunctionNode.java
diff options
context:
space:
mode:
authorJon Bratseth <bratseth@yahoo-inc.com>2017-11-28 21:35:59 +0100
committerJon Bratseth <bratseth@yahoo-inc.com>2017-11-28 21:35:59 +0100
commitb5ffe229474223844c150e99d24ca618e5e9f8dd (patch)
tree7c9ac3da58ff567fae79019bc688ba2a4e4d904c /searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/rule/TensorFunctionNode.java
parent1d6791e6fa004ae80e85dbc6a6c7c2e4b8037a4f (diff)
Complete prototype TensorFlow mapping
Diffstat (limited to 'searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/rule/TensorFunctionNode.java')
-rw-r--r--searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/rule/TensorFunctionNode.java3
1 files changed, 3 insertions, 0 deletions
diff --git a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/rule/TensorFunctionNode.java b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/rule/TensorFunctionNode.java
index ce21e132980..ab5f1e7191d 100644
--- a/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/rule/TensorFunctionNode.java
+++ b/searchlib/src/main/java/com/yahoo/searchlib/rankingexpression/rule/TensorFunctionNode.java
@@ -30,6 +30,9 @@ public class TensorFunctionNode extends CompositeNode {
this.function = function;
}
+ /** Returns the tensor function wrapped by this */
+ public TensorFunction function() { return function; }
+
@Override
public List<ExpressionNode> children() {
return function.functionArguments().stream()