BidirectionalSequenceLSTM op: Fixes VTS test for merge_outputs

Bug: 123644584

Test: VtsHalNeuralnetworksV1_xTargetTest
Change-Id: I8829fd47094ca4af05e5d60da917499ce61acc5e
Merged-In: I8829fd47094ca4af05e5d60da917499ce61acc5e
(cherry picked from commit a8f33f7dbf)
This commit is contained in:
Viet Dang
2019-03-28 17:22:56 +00:00
parent 1e447c7528
commit 723ff2dbfd

View File

@@ -508,14 +508,15 @@ 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 ||
for (const size_t outOprand : operation.outputs) { operation.type == OperationType::BIDIRECTIONAL_SEQUENCE_RNN) {
if (operand == outOprand) { for (const size_t outOprand : operation.outputs) {
return true; if (operand == outOprand) {
} return true;
} }
}
} }
} }
return false; return false;