summaryrefslogtreecommitdiffstats
path: root/model-integration/src/main/java/ai/vespa/rankingexpression/importer/onnx/TensorConverter.java
diff options
context:
space:
mode:
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.java8
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; }
}