diff options
Diffstat (limited to 'vespajlib/src/main/java/com/yahoo/tensor/functions/ConstantTensor.java')
-rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/functions/ConstantTensor.java | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/ConstantTensor.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/ConstantTensor.java index 14ed38718ce..4a6d656142f 100644 --- a/vespajlib/src/main/java/com/yahoo/tensor/functions/ConstantTensor.java +++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/ConstantTensor.java @@ -3,6 +3,7 @@ package com.yahoo.tensor.functions; import com.google.common.annotations.Beta; import com.yahoo.tensor.Tensor; +import com.yahoo.tensor.TensorType; import com.yahoo.tensor.evaluation.EvaluationContext; import java.util.Collections; @@ -27,10 +28,10 @@ public class ConstantTensor extends PrimitiveTensorFunction { } @Override - public List<TensorFunction> functionArguments() { return Collections.emptyList(); } + public List<TensorFunction> arguments() { return Collections.emptyList(); } @Override - public TensorFunction replaceArguments(List<TensorFunction> arguments) { + public TensorFunction withArguments(List<TensorFunction> arguments) { if ( arguments.size() != 1) throw new IllegalArgumentException("ConstantTensor must have 0 arguments, got " + arguments.size()); return this; @@ -40,6 +41,9 @@ public class ConstantTensor extends PrimitiveTensorFunction { public PrimitiveTensorFunction toPrimitive() { return this; } @Override + public TensorType type(EvaluationContext context) { return constant.type(); } + + @Override public Tensor evaluate(EvaluationContext context) { return constant; } @Override |