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

Commit ea551fd1 authored by Viet Dang's avatar Viet Dang
Browse files

BidirectionalSequenceLSTM op: Fixes VTS test for merge_outputs 

Bug: 123644584

Test: VtsHalNeuralnetworksV1_xTargetTest
Change-Id: I8829fd47094ca4af05e5d60da917499ce61acc5e
Merged-In: I8829fd47094ca4af05e5d60da917499ce61acc5e
(cherry picked from commit a8f33f7d)
parent b1a14404

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;