From 673939fd24ee0d9c18bb714d418552ec1233d0fa Mon Sep 17 00:00:00 2001
From: Koan-Sin Tan <koansin.tan@gmail.com>
Date: Sat, 11 May 2019 09:42:21 +0800
Subject: [PATCH] make elementwise ops work

---
 tensorflow/lite/delegates/gpu/common/model_builder.cc | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/tensorflow/lite/delegates/gpu/common/model_builder.cc b/tensorflow/lite/delegates/gpu/common/model_builder.cc
index 32b7506a82c..b616a72378c 100644
--- a/tensorflow/lite/delegates/gpu/common/model_builder.cc
+++ b/tensorflow/lite/delegates/gpu/common/model_builder.cc
@@ -1240,8 +1240,7 @@ class ElementwiseOperationParser : public TFLiteOperationParser {
       TfLiteSubParams* tf_options = nullptr;
       RETURN_IF_ERROR(RetrieveBuiltinData(tflite_node, &tf_options));
       RETURN_IF_ERROR(CheckActivationSupported(tf_options->activation));
-    }
-    if (!IsOneArgumentOperation()) {
+    } else if (!IsOneArgumentOperation()) {
       return InvalidArgumentError("Incorrect operation type passed");
     }