Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 80d016b7 authored by Viet Dang's avatar Viet Dang Committed by android-build-merger
Browse files

Merge "BidirectionalSequenceLSTM op: Fixes VTS test for merge_outputs" am: 77f7fdfb 

am: aea0439a

Change-Id: I09f5e1be87170b1403f6431d54702d09cd31b891
parents 7c2f483f aea0439a

neuralnetworks/1.2/vts/functional/ValidateModel.cpp

+8 −7
Original line number Original line Diff line number Diff line
@@ -508,9 +508,10 @@ static bool removeOperandSkip(size_t operand, const Model& model) { 
                }

                }

            }

            }

        }

        }

        // BIDIRECTIONAL_SEQUENCE_RNN can have either on or two outputs

        // BIDIRECTIONAL_SEQUENCE_LSTM and BIDIRECTIONAL_SEQUENCE_RNN can have

        // depending on a mergeOutputs parameter

        // either one or two outputs depending on their mergeOutputs parameter.

        if (operation.type == OperationType::BIDIRECTIONAL_SEQUENCE_RNN) {

        if (operation.type == OperationType::BIDIRECTIONAL_SEQUENCE_LSTM ||

            operation.type == OperationType::BIDIRECTIONAL_SEQUENCE_RNN) {

          for (const size_t outOprand : operation.outputs) {

          for (const size_t outOprand : operation.outputs) {

            if (operand == outOprand) {

            if (operand == outOprand) {

              return true;

              return true;