Commit Graph

5 Commits (d9270e34d1505294fcaa268f095f469233e9eb5d)

Author SHA1 Message Date
Yiqun Liu 3008fa1261
Add the CUDA kernel for beam_search op (#15020)
6 years ago
frankwhzhang c9a653820b fix label_pos ,add test_layers.py, test=develop
6 years ago
frankwhzhang a672b291e5 fix code style, test=develop
6 years ago
frankwhzhang 97de98cd0a update bpr_loss op code, test=develop
7 years ago
frankwhzhang 570d89ec84 add bpr_loss operator , test=develop
7 years ago