4 Commits

Author SHA1 Message Date
Fangjun Kuang
cafd06e909 Fix test failures for torch 1.8.0 2021-12-06 10:42:01 +08:00
Fangjun Kuang
3d38f7bd31 Add GPU tests. 2021-12-03 17:22:42 +08:00
Fangjun Kuang
2c7547e1b7 Add projection support to LayerNormLSTMCell. 2021-12-03 16:47:40 +08:00
Fangjun Kuang
1d004ca966 Apply layer normalization to the output of each gate in GRU. 2021-12-03 14:59:19 +08:00