diff options
author | Lester Solbakken <lesters@oath.com> | 2021-04-08 11:24:52 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2021-04-08 11:24:52 +0200 |
commit | 049e9a325c8142958909d0464da12a56e5a8f638 (patch) | |
tree | 31d857ec4a5ad3415464e480ae473c39224623b2 /vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java | |
parent | bccd68f8f9a7eb0830d136f8b034ae4f40cc819c (diff) |
Add bfloat16 and int8 tensor cell types in Java
Diffstat (limited to 'vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java')
-rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java | 4 |
1 files changed, 4 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 f608aead347..be5f4143f54 100644 --- a/vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java +++ b/vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java @@ -533,6 +533,10 @@ public class MixedTensor implements Tensor { b.append(getDouble(subspaceIndex, index, tensor)); else if (tensor.type().valueType() == TensorType.Value.FLOAT) b.append(getDouble(subspaceIndex, index, tensor)); // TODO: Really use floats + else if (tensor.type().valueType() == TensorType.Value.BFLOAT16) + b.append(getDouble(subspaceIndex, index, tensor)); + else if (tensor.type().valueType() == TensorType.Value.INT8) + b.append(getDouble(subspaceIndex, index, tensor)); else throw new IllegalStateException("Unexpected value type " + type.valueType()); |