summaryrefslogtreecommitdiffstats
path: root/vespajlib/src/main/java/com/yahoo/tensor/TensorType.java
diff options
context:
space:
mode:
Diffstat (limited to 'vespajlib/src/main/java/com/yahoo/tensor/TensorType.java')
-rw-r--r--vespajlib/src/main/java/com/yahoo/tensor/TensorType.java6
1 files changed, 6 insertions, 0 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/TensorType.java b/vespajlib/src/main/java/com/yahoo/tensor/TensorType.java
index acba9eafd71..bb7a976af9b 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/TensorType.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/TensorType.java
@@ -24,9 +24,15 @@ import java.util.stream.Collectors;
*/
public class TensorType {
+ public enum ValueType { DOUBLE, FLOAT};
+
/** The empty tensor type - which is the same as a double */
public static final TensorType empty = new TensorType(Collections.emptyList());
+ private final ValueType valueType = ValueType.DOUBLE;
+
+ public final ValueType valueType() { return valueType; }
+
/** Sorted list of the dimensions of this */
private final ImmutableList<Dimension> dimensions;