tequilaOS/platform_hardware_interfaces
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge "BidirectionalSequenceLSTM op: Fixes VTS test for merge_outputs"

Browse source
This commit is contained in:
Treehugger Robot 2019-04-09 15:00:55 +00:00 committed by Gerrit Code Review
commit 77f7fdfb70
1 changed files with 8 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
neuralnetworks/1.2/vts/functional/ValidateModel.cpp
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
// depending on a mergeOutputs parameter
if (operation.type == OperationType::BIDIRECTIONAL_SEQUENCE_RNN) {
for (const size_t outOprand : operation.outputs) {
if (operand == outOprand) {
return true;
}
// 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;
}
}
}
}
return false;