diff options
Diffstat (limited to 'model-integration/src/main/java/ai/vespa/rankingexpression/importer/onnx/TensorConverter.java')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/rankingexpression/importer/onnx/TensorConverter.java | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/onnx/TensorConverter.java b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/onnx/TensorConverter.java index f8c7dc15857..c8d7392bb8d 100644 --- a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/onnx/TensorConverter.java +++ b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/onnx/TensorConverter.java @@ -66,13 +66,13 @@ class TensorConverter { } private static class RawBoolValues extends RawValues { - private final IntBuffer values; + private final ByteString values; private final int size; RawBoolValues(Onnx.TensorProto tensorProto) { - values = bytes(tensorProto).asIntBuffer(); - size = values.remaining(); + values = tensorProto.getRawData(); + size = values.size(); } - @Override double get(int i) { return values.get(i); } + @Override double get(int i) { return values.byteAt(i) == 0 ? 0.0 : 1.0; } @Override int size() { return size; } } |