BidirectionalSequenceLSTM op: Fixes VTS test for merge_outputs

Bug: 123644584

Test: VtsHalNeuralnetworksV1_xTargetTest
Change-Id: I8829fd47094ca4af05e5d60da917499ce61acc5e
This commit is contained in:
Viet Dang
2019-03-28 17:22:56 +00:00
parent 1b3049e56c
commit a8f33f7dbf

View File

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