- 28 6月, 2021 1 次提交
-
-
由 Qi Li 提交于
* [ROCM] fix RNN miopen as weight need to permuted, test=develop * [ROCM] fix data share when is_test, test=develop * update, test=develop
-
- 19 3月, 2021 1 次提交
-
-
由 ronnywang 提交于
-
- 27 11月, 2020 1 次提交
-
-
由 Jack Zhou 提交于
Add eigen gru and fix the dropout bug in the rnn
-
- 20 11月, 2020 1 次提交
-
-
由 Jack Zhou 提交于
* add lstm, simple rnn op kernel * fix the test_lstm for the rnn op * change func name * fix forward postprocess bug * add gru forward, backward code * remove unittest.skipIf; use a big rnn op instead of combination op * fix input doesn't have gradient bug * add eigen lstm forward, backward Co-authored-by: Nwawltor <fangzeyang0904@hotmail.com>
-