commit c8fc2114f8b9b8ecb91f27a973a7a8056c2e030c
author Raj Mojumder <rmojumder@google.com> Wed Apr 10 22:40:43 2019 -0700
committer android-build-merger <android-build-merger@google.com> Wed Apr 10 22:40:43 2019 -0700
tree 5f9d8fa71b3e2c43c200a0bf0f3f093663eb7010
parent 22e035ad725a6f7afa3a3e9427724184a1abe1f5
parent 092c9f5e80bc431f5d7b55c7b540c7b661b6d646

Merge "NFC: Update OWNERS for default/" into qt-dev
am: 092c9f5e80

Change-Id: If856e72efa7e54b456c1b8d848ac11a596be5584

neuralnetworks/1.2/vts/functional/ValidateModel.cpp

diff --git a/neuralnetworks/1.2/vts/functional/ValidateModel.cpp b/neuralnetworks/1.2/vts/functional/ValidateModel.cpp
index 2988211..f87ca89 100644
--- a/neuralnetworks/1.2/vts/functional/ValidateModel.cpp
+++ b/neuralnetworks/1.2/vts/functional/ValidateModel.cpp
@@ -508,14 +508,15 @@
                 }
             }
         }
-        // BIDIRECTIONAL_SEQUENCE_RNN can have either on or two outputs
-        // depending on a mergeOutputs parameter
-        if (operation.type == OperationType::BIDIRECTIONAL_SEQUENCE_RNN) {
-            for (const size_t outOprand : operation.outputs) {
-                if (operand == outOprand) {
-                    return true;
-                }
+        // BIDIRECTIONAL_SEQUENCE_LSTM and BIDIRECTIONAL_SEQUENCE_RNN can have
+        // either one or two outputs depending on their mergeOutputs parameter.
+        if (operation.type == OperationType::BIDIRECTIONAL_SEQUENCE_LSTM ||
+            operation.type == OperationType::BIDIRECTIONAL_SEQUENCE_RNN) {
+          for (const size_t outOprand : operation.outputs) {
+            if (operand == outOprand) {
+              return true;
             }
+          }
         }
     }
     return false;