diff options
author | Lester Solbakken <lesters@oath.com> | 2021-04-16 14:54:33 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2021-04-16 14:54:33 +0200 |
commit | 2b39900fd096ff8b0a7f351164bd7f4f73bda501 (patch) | |
tree | 36bcd7521b1cceff4bd2bdafef92c0b4abebdf96 | |
parent | 2c930ee0bf3137bf8e355f95a58e4255b046989e (diff) |
Wire inn tensor cell type resolving for map in Java
-rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/functions/Map.java | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/Map.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/Map.java index 5eef1d0bec4..77c23f2e603 100644 --- a/vespajlib/src/main/java/com/yahoo/tensor/functions/Map.java +++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/Map.java @@ -4,6 +4,7 @@ package com.yahoo.tensor.functions; import com.yahoo.tensor.Tensor; import com.yahoo.tensor.TensorAddress; import com.yahoo.tensor.TensorType; +import com.yahoo.tensor.TypeResolver; import com.yahoo.tensor.evaluation.EvaluationContext; import com.yahoo.tensor.evaluation.Name; import com.yahoo.tensor.evaluation.TypeContext; @@ -31,7 +32,9 @@ public class Map<NAMETYPE extends Name> extends PrimitiveTensorFunction<NAMETYPE this.mapper = mapper; } - public static TensorType outputType(TensorType inputType) { return inputType; } + public static TensorType outputType(TensorType inputType) { + return TypeResolver.map(inputType); + } public TensorFunction<NAMETYPE> argument() { return argument; } public DoubleUnaryOperator mapper() { return mapper; } |