Build tensors purely with floats

1 files changed, 15 insertions, 0 deletions

diff --git a/vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java b/vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java
index c06cb2a0986..95f64cec0c1 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java
@@ -193,6 +193,11 @@ public class MixedTensor implements Tensor {
         }
 
         @Override
+        public Tensor.Builder cell(float value, long... labels) {
+            return cell((double)value, labels);
+        }
+
+        @Override
         public Tensor.Builder cell(double value, long... labels) {
             throw new UnsupportedOperationException("Not implemented.");
         }
@@ -236,6 +241,11 @@ public class MixedTensor implements Tensor {
         }
 
         @Override
+        public Tensor.Builder cell(TensorAddress address, float value) {
+            return cell(address, (double)value);
+        }
+
+        @Override
         public Tensor.Builder cell(TensorAddress address, double value) {
             TensorAddress sparsePart = index.sparsePartialAddress(address);
             long denseOffset = index.denseOffset(address);
@@ -293,6 +303,11 @@ public class MixedTensor implements Tensor {
         }
 
         @Override
+        public Tensor.Builder cell(TensorAddress address, float value) {
+            return cell(address, (double)value);
+        }
+
+        @Override
         public Tensor.Builder cell(TensorAddress address, double value) {
             cells.put(address, value);
             trackBounds(address);