Add parallel linking support for BIDIRECTIONAL_SEQUENCE_LSTM
Previoulsy, BIDIRECTIONAL_SEQUENCE_LSTM only supported cross-linking
mode, i.e. passing the regular and the auxiliary input to both the
forward and backward networks in the next BIDIRECTIONAL_SEQUENCE_LSTM op:
AUX_INPUT (AUX_INPUT_REVERSED)
| |
INPUT | (INPUT_R'D.)|
| | | |
-----------------------
| \ / \ / |
| FW_LSTM BW_LSTM |
-----------------------
| |
FW_OUT BW_OUT
This CL adds support for parallel linking of bidirectional sequence
LSTMs, where the regular input is passed only to the forward network and
the auxiliary input is passed only to the backward network:
INPUT (AUX_INPUT_REVERSED)
| |
---------------------
| FW_LSTM BW_LSTM |
---------------------
| |
FW_OUT BW_OUT
Fix: 138653129
Test: NNTest_static and VtsHalNeuralnetworksV1_3TargetTest
Change-Id: Iba255c6377aa8f74d78e09c44c483ec406fe6539
Merged-In: Iba255c6377aa8f74d78e09c44c483ec406fe6539
(cherry picked from commit e613aa6e784fba2cb8725c7025085a79ac48c33c)
5 files changed