1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
// Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package ai.vespa.rankingexpression.importer.operations;
import com.yahoo.searchlib.rankingexpression.Reference;
import com.yahoo.searchlib.rankingexpression.evaluation.Value;
import ai.vespa.rankingexpression.importer.DimensionRenamer;
import ai.vespa.rankingexpression.importer.OrderedTensorType;
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.TensorFunction;
import java.util.List;
import java.util.Optional;
public class Const extends IntermediateOperation {
private final AttributeMap attributeMap;
public Const(String modelName,
String nodeName,
List<IntermediateOperation> inputs,
AttributeMap attributeMap,
OrderedTensorType type) {
super(modelName, nodeName, inputs);
this.attributeMap = attributeMap;
this.type = type.rename(vespaName() + "_");
setConstantValue(value());
}
@Override
protected OrderedTensorType lazyGetType() {
return type;
}
@Override
public Optional<TensorFunction> function() {
if (function == null) {
function = lazyGetFunction();
}
return Optional.ofNullable(function);
}
@Override
protected TensorFunction lazyGetFunction() {
ExpressionNode expressionNode;
if (type.type().rank() == 0 && getConstantValue().isPresent()) {
expressionNode = new ConstantNode(getConstantValue().get().asDoubleValue());
} else {
expressionNode = new ReferenceNode(Reference.simple("constant", vespaName()));
}
return new TensorFunctionNode.ExpressionTensorFunction(expressionNode);
}
/** Constant names are prefixed by "modelName_" to avoid name conflicts between models */
@Override
public String vespaName() {
return modelName + "_" + super.vespaName();
}
@Override
public void addDimensionNameConstraints(DimensionRenamer renamer) {
addConstraintsFrom(type, renamer);
}
@Override
public void renameDimensions(DimensionRenamer renamer) {
super.renameDimensions(renamer);
setConstantValue(value());
}
@Override
public boolean isConstant() {
return true;
}
private Value value() {
Optional<Value> value = attributeMap.get("value", type);
if ( ! value.isPresent()) {
throw new IllegalArgumentException("Node '" + name + "' of type " +
"const has missing or non-recognized 'value' attribute");
}
return value.get();
}
@Override
public Const withInputs(List<IntermediateOperation> inputs) {
return new Const(modelName(), name(), inputs, attributeMap, type);
}
@Override
public String operationName() { return "Const"; }
@Override
public String toString() {
return "Const(" + type + ")";
}
@Override
public String toFullString() {
return "\t" + type + ":\tConst(" + getConstantValue().get() + ")";
}
}
|
