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

Commit 014dd02f authored by Viet Dang's avatar Viet Dang Committed by Android (Google) Code Review
Browse files

Merge "BidirectionalSequenceLSTM op: Fixes VTS test for merge_outputs" into qt-dev

parents eeb44901 ea551fd1

neuralnetworks/1.2/vts/functional/ValidateModel.cpp

+4 −3
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;