|
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
|