From db1e43e3335e190db1a27e0e74f104d7bc5471dd Mon Sep 17 00:00:00 2001 From: Asim Shankar Date: Mon, 28 Aug 2017 13:34:27 -0700 Subject: Java: Support multi-dimensional STRING tensors. Fixes #8531 PiperOrigin-RevId: 166749470 --- .../contrib/android/TensorFlowInferenceInterface.java | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'tensorflow/contrib/android') diff --git a/tensorflow/contrib/android/java/org/tensorflow/contrib/android/TensorFlowInferenceInterface.java b/tensorflow/contrib/android/java/org/tensorflow/contrib/android/TensorFlowInferenceInterface.java index a75ec45e0b..6389ef1f5d 100644 --- a/tensorflow/contrib/android/java/org/tensorflow/contrib/android/TensorFlowInferenceInterface.java +++ b/tensorflow/contrib/android/java/org/tensorflow/contrib/android/TensorFlowInferenceInterface.java @@ -267,6 +267,25 @@ public class TensorFlowInferenceInterface { addFeed(inputName, Tensor.create(DataType.UINT8, dims, ByteBuffer.wrap(src))); } + /** + * Copy a byte sequence into the input Tensor with name {@link inputName} as a string-valued + * scalar tensor. In the TensorFlow type system, a "string" is an arbitrary sequence of + * bytes, not a Java {@code String} (which is a sequence of characters). + */ + public void feedString(String inputName, byte[] src) { + addFeed(inputName, Tensor.create(src)); + } + + /** + * Copy an array of byte sequences into the input Tensor with name {@link inputName} as a + * string-valued one-dimensional tensor (vector). In the TensorFlow type system, a "string" + * is an arbitrary sequence of bytes, not a Java {@code String} (which is a sequence of + * characters). + */ + public void feedString(String inputName, byte[][] src) { + addFeed(inputName, Tensor.create(src)); + } + // Methods for taking a native Tensor and filling it with src from Java native IO buffers. /** -- cgit v1.2.3