diff options
-rw-r--r-- | tensorflow/contrib/android/java/org/tensorflow/contrib/android/TensorFlowInferenceInterface.java | 2 |
1 files changed, 1 insertions, 1 deletions
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 2bc6acd957..76a429476f 100644 --- a/tensorflow/contrib/android/java/org/tensorflow/contrib/android/TensorFlowInferenceInterface.java +++ b/tensorflow/contrib/android/java/org/tensorflow/contrib/android/TensorFlowInferenceInterface.java @@ -467,7 +467,7 @@ public class TensorFlowInferenceInterface { private Tensor getTensor(String outputName) { int i = 0; for (String n : fetchNames) { - if (n == outputName) { + if (n.equals(outputName)) { return fetchTensors.get(i); } i++; |