diff options
author | Arne Juul <arnej@verizonmedia.com> | 2021-04-15 09:38:42 +0000 |
---|---|---|
committer | Arne Juul <arnej@verizonmedia.com> | 2021-04-15 10:42:36 +0000 |
commit | 1e2b18e883daa40768a41607295b350bc0990c0d (patch) | |
tree | 8312ef95b04956eb3930324363945fff549fa61e /vespajlib/src | |
parent | d26f4ddf2c8720f05d793f4a4ef4119b9d3a6485 (diff) |
wire in TypeResolver.reduce
Diffstat (limited to 'vespajlib/src')
-rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/functions/Reduce.java | 9 |
1 files changed, 2 insertions, 7 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/Reduce.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/Reduce.java index 797b890dcb9..263bb6c9025 100644 --- a/vespajlib/src/main/java/com/yahoo/tensor/functions/Reduce.java +++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/Reduce.java @@ -6,6 +6,7 @@ import com.yahoo.tensor.IndexedTensor; 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; @@ -60,13 +61,7 @@ public class Reduce<NAMETYPE extends Name> extends PrimitiveTensorFunction<NAMET } public static TensorType outputType(TensorType inputType, List<String> reduceDimensions) { - TensorType.Builder b = new TensorType.Builder(inputType.valueType()); - if (reduceDimensions.isEmpty()) return b.build(); // means reduce all - for (TensorType.Dimension dimension : inputType.dimensions()) { - if ( ! reduceDimensions.contains(dimension.name())) - b.dimension(dimension); - } - return b.build(); + return TypeResolver.reduce(inputType, reduceDimensions); } public TensorFunction<NAMETYPE> argument() { return argument; } |