diff options
Diffstat (limited to 'tensorflow/java/src/main/java/org/tensorflow/Tensor.java')
| -rw-r--r-- | tensorflow/java/src/main/java/org/tensorflow/Tensor.java | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/java/src/main/java/org/tensorflow/Tensor.java b/tensorflow/java/src/main/java/org/tensorflow/Tensor.java index 38bb55e59f..8987253768 100644 --- a/tensorflow/java/src/main/java/org/tensorflow/Tensor.java +++ b/tensorflow/java/src/main/java/org/tensorflow/Tensor.java @@ -595,7 +595,7 @@ public final class Tensor<T> implements AutoCloseable { } private static int elemByteSize(DataType dataType) { - int size = dataType.sizeInBytes(); + int size = dataType.byteSize(); if (size < 0) { throw new IllegalArgumentException("STRING tensors do not have a fixed element size"); } |