commit e60b0e06c30c518b1f0d03de03522a7e5c05b023
author Steven Moreland <smoreland@google.com> Tue Apr 16 10:47:16 2019 -0700
committer android-build-merger <android-build-merger@google.com> Tue Apr 16 10:47:16 2019 -0700
tree 7ceed4a4bd8dec8e888b6f804f6421f090c060c8
parent 483255ef3c3adb592be038c126dd004057006007
parent 86eb7fc62bd138c746e20171124abaf9b975006c

Compatibility_matrix: s/current/4/
am: 86eb7fc62b

Change-Id: Ie2d6e9c088c5c50024e5ed76ef3008125ea20acc

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;