search

ARM-software / armnn

Arm NN ML Software. The code here is a read-only mirror of https://review.mlplatform.org/admin/repos/ml/armnn
https://developer.arm.com/products/processors/machine-learning/arm-nn
MIT License
1.14k stars 307 forks source link

UnidirectionalSequenceLSTMLayer's inferShape is wrong written! #780

Open clay001 opened 6 days ago

clay001 commented 6 days ago

file src/armnn/layers/UnidirectionalSequenceLstmLayer.cpp the line 360, ValidateAndCopyShape function, if you do not pass the OutputSlotIndex param, the default will set the inferredShapes to the first outputSlot which is not correct, the real output should be the last outputSlot.

Can some body help me to check my concern?