diff options
author | Jon Bratseth <bratseth@verizonmedia.com> | 2019-04-26 14:24:17 +0200 |
---|---|---|
committer | Jon Bratseth <bratseth@verizonmedia.com> | 2019-04-26 14:24:17 +0200 |
commit | ae5d5e058f1bb2fd197886ac374ce807065fdb77 (patch) | |
tree | 2966fda95d45f68ccf212e9fe8884528b7ce23f6 /vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java | |
parent | 94b4b3ad837f9d3f9d43b158c4de8475ff2c2a2d (diff) |
Build tensors purely with floats
Diffstat (limited to 'vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java')
-rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java | 15 |
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); |