| commit | 29f7bfaf16d9f19f6c0197bbe34a9fc5ff79bd1d | [log] [tgz] |
|---|---|---|
| author | Asim Shankar <ashankar@google.com> | Fri Feb 10 18:06:47 2017 -0800 |
| committer | TensorFlower Gardener <gardener@tensorflow.org> | Fri Feb 10 18:25:48 2017 -0800 |
| tree | faa88cb0b3c36ea54b372574b6eee7f1fb32bab4 | |
| parent | 63ad7054ad3fc223f94bbb5d7a203b6d08eceb07 [diff] |
Java: Bugfix: Handle UINT8 correctly in elemByteSize. Change: 147221491
diff --git a/tensorflow/java/src/main/java/org/tensorflow/Tensor.java b/tensorflow/java/src/main/java/org/tensorflow/Tensor.java index aeb53fb..efd6c81 100644 --- a/tensorflow/java/src/main/java/org/tensorflow/Tensor.java +++ b/tensorflow/java/src/main/java/org/tensorflow/Tensor.java
@@ -497,6 +497,8 @@ private static int elemByteSize(DataType dataType) { switch (dataType) { + case UINT8: + return 1; case FLOAT: case INT32: return 4;