Commit Graph

5 Commits (8fc0fc314a51cfea0579d0ac058349b051e688d4)

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
7 years ago
frankwhzhang a672b291e5 fix code style, test=develop
7 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