Evolution-x/hardware_interfaces
1
0
Fork 0
mirror of https://github.com/Evolution-X/hardware_interfaces synced 2026-02-01 16:50:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Viet Dang
2019-04-18 10:22:13 +00:00
committed by Android (Google) Code Review
parent eeb449016d ea551fd12a
commit 014dd02f6e
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
neuralnetworks/1.2/vts/functional/ValidateModel.cpp
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 // BIDIRECTIONAL_SEQUENCE_LSTM and BIDIRECTIONAL_SEQUENCE_RNN can have either one or two
// depending on a mergeOutputs parameter // outputs depending on their mergeOutputs parameter.
if (operation.type == OperationType::BIDIRECTIONAL_SEQUENCE_RNN) { if (operation.type == OperationType::BIDIRECTIONAL_SEQUENCE_LSTM ||
operation.type == OperationType::BIDIRECTIONAL_SEQUENCE_RNN) {
for (const size_t outOprand : operation.outputs) { for (const size_t outOprand : operation.outputs) {
if (operand == outOprand) { if (operand == outOprand) {
return true; return true;