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

BidirectionalSequenceLSTM op: Fixes VTS test for merge_outputs

Browse Source 
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 b1a144041e
commit ea551fd12a
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;