summaryrefslogtreecommitdiffstats
path: root/vespajlib/src/main/java/com/yahoo/tensor/serialization/DenseBinaryFormat.java
diff options
context:
space:
mode:
Diffstat (limited to 'vespajlib/src/main/java/com/yahoo/tensor/serialization/DenseBinaryFormat.java')
-rw-r--r--vespajlib/src/main/java/com/yahoo/tensor/serialization/DenseBinaryFormat.java17
1 files changed, 2 insertions, 15 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/serialization/DenseBinaryFormat.java b/vespajlib/src/main/java/com/yahoo/tensor/serialization/DenseBinaryFormat.java
index 4d4d947557d..c3284131be0 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/serialization/DenseBinaryFormat.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/serialization/DenseBinaryFormat.java
@@ -34,7 +34,7 @@ public class DenseBinaryFormat implements BinaryFormat {
private void encodeDimensions(GrowableByteBuffer buffer, IndexedTensor tensor) {
buffer.putInt1_4Bytes(tensor.type().dimensions().size());
for (int i = 0; i < tensor.type().dimensions().size(); i++) {
- encodeString(buffer, tensor.type().dimensions().get(i).name());
+ buffer.putUtf8String(tensor.type().dimensions().get(i).name());
buffer.putInt1_4Bytes(tensor.dimensionSizes().size(i));
}
}
@@ -50,12 +50,6 @@ public class DenseBinaryFormat implements BinaryFormat {
}
}
- private void encodeString(GrowableByteBuffer buffer, String value) {
- byte[] stringBytes = Utf8.toBytes(value);
- buffer.putInt1_4Bytes(stringBytes.length);
- buffer.put(stringBytes);
- }
-
@Override
public Tensor decode(TensorType type, GrowableByteBuffer buffer) {
DimensionSizes sizes = decodeDimensionSizes(type, buffer);
@@ -74,7 +68,7 @@ public class DenseBinaryFormat implements BinaryFormat {
for (int i = 0; i < dimensionCount; i++) {
TensorType.Dimension expectedDimension = type.dimensions().get(i);
- String encodedName = decodeString(buffer);
+ String encodedName = buffer.getUtf8String();
int encodedSize = buffer.getInt1_4Bytes();
if ( ! expectedDimension.name().equals(encodedName))
@@ -95,11 +89,4 @@ public class DenseBinaryFormat implements BinaryFormat {
builder.cellByDirectIndex(i, buffer.getDouble());
}
- private String decodeString(GrowableByteBuffer buffer) {
- int stringLength = buffer.getInt1_4Bytes();
- byte[] stringBytes = new byte[stringLength];
- buffer.get(stringBytes);
- return Utf8.toString(stringBytes);
- }
-
}