- 19 2月, 2021 1 次提交
-
-
由 Wojciech Uss 提交于
* Modify relu native implementation * fix GPU performance
-
- 15 1月, 2021 1 次提交
-
-
由 wawltor 提交于
* fix the rnn mask memory bug for out of read * update the code for the rnn
-
- 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>
-