summaryrefslogtreecommitdiff
path: root/nn/runtime/test/specs/V1_3/bidirectional_sequence_lstm.mod.py
AgeCommit message (Collapse)Author
2020-02-18Add parallel linking support for BIDIRECTIONAL_SEQUENCE_LSTMLev Proleev
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