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

Commit aea0439a 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

Change-Id: Ieccbe708e378de8716d4961302b07be2d4fa2cf4
parents 60cb34be 77f7fdfb

neuralnetworks/1.2/vts/functional/ValidateModel.cpp

+8 −7
Original line number 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

        // depending on a mergeOutputs parameter

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

        // 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;