aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
-rw-r--r--tensorflow/contrib/android/java/org/tensorflow/contrib/android/TensorFlowInferenceInterface.java2
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++;