This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Fix LSTM and GRU layers gradient calculations - #18203

Merged

pengzhao-intel merged 6 commits into

apache:masterfrom

bgawrych:rnn_fix

May 14, 2020

Commits on May 12, 2020

Fix input gradient calculation for bidirectional LSTM
Bartlomiej Gawrych
committed
be2e87c
Fix gradient calculation for GRU
Bartlomiej Gawrych
committed
624e530
Enable tests for GRU and LSTM gradients
Bartlomiej Gawrych
committed
7bb38c4
Fix comments
Bartlomiej Gawrych
committed
201f671
Change loop iteration deduction
Bartlomiej Gawrych
committed
df6e90c
Add more test cases for fused rnn layers
Bartlomiej Gawrych
committed
34d8947